Update app.py

app.py

@@ -12,7 +12,7 @@ else:
     print("Using CPU")
 
 tokenizer = T5Tokenizer.from_pretrained("roborovski/superprompt-v1")
-model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", torch_dtype=torch.float16)
+model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", device_map="auto", torch_dtype="auto")
 model.to(device)
 
 def generate(your_prompt, task_prefix, max_new_tokens, repetition_penalty, temperature, model_precision_type, top_p, top_k, seed):