Update app.py

app.py

@@ -13,7 +13,7 @@ else:
     print("Using CPU")
 
 tokenizer = T5Tokenizer.from_pretrained("roborovski/superprompt-v1")
-model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1")
+model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", torch_dtype=torch.float16)
 model.to(device)
 
 @spaces.GPU()