Update app.py

app.py

@@ -5,8 +5,10 @@ import gradio as gr
 warnings.filterwarnings('ignore')
 import os
 
+from huggingface_hub import from_pretrained_keras
+
+model = from_pretrained_keras("soufyane/gemma_data_science")
 
-model = keras_nlp.models.CausalLM.from_preset(f"hf://soufyane/gemma_data_science")
 
 def process_text_gemma(input_text):
     response = model.generate(f"question: {input_text}", max_length=256)