"Update space"

Signed-off-by: Emmanuel Frimpong Asante <[email protected]>

app.py

@@ -7,6 +7,7 @@ import torch  # Import torch here
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from huggingface_hub import login
 import os
+import spaces
 
 # Ensure the HF token is set
 tok = os.getenv('HF_Token')
@@ -53,6 +54,7 @@ def predict(image):
     return f"Chicken is {status}, the disease it has is {name}, the recommended medication is {recom}"
 
 
+@spaces.GPU(duration=200)
 def chat_response(user_input):
     inputs = llama_tokenizer(user_input, return_tensors='pt')
     outputs = llama_model.generate(inputs['input_ids'], max_length=500, do_sample=True)
@@ -88,4 +90,4 @@ interface = gr.Interface(
 )
 
 # Launch the interface
-interface.launch(debug=True, share=True)
+interface.launch(debug=True)