Update app.py

app.py

@@ -16,6 +16,7 @@ else:
 model.to(device)
 
 def generate(
+    precision_model,
     system_prompt,
     prompt,
     max_new_tokens,
@@ -23,8 +24,7 @@ def generate(
     temperature,
     top_p,
     top_k,
-    seed,
-    precision_model
+    seed
 ):
 
     model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", torch_dtype=precision_model)
@@ -86,7 +86,7 @@ examples = [
 
 gr.Interface(
     fn=generate,
-    inputs=[prompt, system_prompt, max_new_tokens, repetition_penalty, temperature, top_p, top_k, seed],
+    inputs=[precision_model, prompt, system_prompt, max_new_tokens, repetition_penalty, temperature, top_p, top_k, seed],
     outputs=gr.Textbox(label="Better Prompt", interactive=True),
     title="SuperPrompt-v1",
     description="Make your prompts more detailed!<br>Model used: https://huggingface.co/roborovski/superprompt-v1<br>Hugging Face Space made by [Nick088](https://linktr.ee/Nick088)",