Update app.py

app.py

@@ -14,11 +14,10 @@ def generate(
     top_k,
     random_seed,
     seed,
-    model_path="roborovski/superprompt-v1",
-    dtype="fp16",
+    precision=torch.float16,
 ):
     tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-small")
-    model = T5ForConditionalGeneration.from_pretrained(model_path, torch_dtype=dtype)
+    model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", torch_dtype=precision)
 
     if torch.cuda.is_available():
         device = "cuda"
@@ -128,8 +127,7 @@ examples = [
         50,
         False,
         42,
-        "roborovski/superprompt-v1",
-        "fp16",
+        torch.float16,
     ]
 ]