Update app.py

app.py

@@ -5,6 +5,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 model_name = "loocorez/reverse-text-warmup"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
+model.to("cuda")
 
 def reverse_text(input_text):
     # Add your model inference logic here