updated requirements.txt

README.md

@@ -4,7 +4,7 @@ emoji: 💬
 colorFrom: yellow
 colorTo: purple
 sdk: gradio
-sdk_version: 5.0.1
+sdk_version: 5.6.0
 app_file: app.py
 pinned: false
 license: mit

app.py

@@ -1,10 +1,12 @@
 import gradio as gr
 import keras_nlp
+print("Modules loaded!")
 
 """
 For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
 """
 
+print("Fetching model...")
 model = keras_nlp.models.GemmaCausalLM.from_preset("kaggle://bhashwar22/gemma-for-finance/keras/gemma-for-finance")
 print("model successfully loaded!")
 
@@ -34,12 +36,14 @@ def respond(
 
     messages += f"Question: {message}\nAnswer: "
 
+    print("Generating response...")
     output = model.generate(
         messages,
         max_tokens=max_tokens,
         temperature=temperature,
         top_p=top_o
     )
+    print("Response generated!")
 
     # Split by "Answer:" from the right and get the last part
     response = output.rsplit("Answer: ", 1)[-1]

requirements.txt

@@ -1 +1,6 @@
-huggingface_hub==0.25.2
+gradio
+huggingface_hub==0.25.2
+keras==3.4.1
+keras_nlp==0.15.1
+transformers
+tensorflow