Update app.py

app.py

@@ -12,6 +12,11 @@ if torch.cuda.is_available():
 else:
     device = "cpu"
     print("Using CPU")
+
+
+model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", torch_dtype=torch.float16)
+
+model.to(device)
     
 
 def generate(