Update app.py

app.py

@@ -13,7 +13,7 @@ else:
 tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-small")
 
 def generate(prompt, model_precision_type, max_new_tokens, repetition_penalty, temperature, top_p, top_k, seed):
-    model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", device_map="auto", torch_dtype=model_precision_type)
+    model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", device_map="auto", torch_dtype=torch.fp16)
     model.to(device)
 
     input_text = f"Expand the following prompt to add more detail: {prompt}"