Update app.py

app.py

@@ -90,7 +90,7 @@ def load_cure_generator():
     try:
         tokenizer = AutoTokenizer.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
         model = AutoModelForCausalLM.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
-        return pipeline("text-generation", model=model, tokenizer=tokenizer, device=-1)
+        return pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)
     except Exception as e:
         logger.error(f"Error loading cure generator: {e}")