Update app.py

app.py

@@ -20,7 +20,6 @@ model.to(device)
     
 
 def generate(
-    precision_model,
     system_prompt,
     prompt,
     max_new_tokens,
@@ -30,10 +29,6 @@ def generate(
     top_k,
     seed
 ):
-
-    model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", torch_dtype=precision_model)
-
-    model.to(device)
     
     input_text = f"{system_prompt}, {prompt}"
     input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to(device)
@@ -59,8 +54,6 @@ def generate(
     return better_prompt
 
 
-precision_model = gr.Radio([('fp32', torch.float32), ('fp16', torch.float16)], value=torch.float16, label="Model Precision Type", info="fp32 is more precised but slower, fp16 is faster and less resource consuming but less pricse")
-
 prompt = gr.Textbox(label="Prompt", interactive=True)
 
 system_prompt = gr.Textbox(label="System Prompt", interactive=True)
@@ -79,7 +72,6 @@ seed = gr.Number(value=42, interactive=True, label="Seed", info="A starting poin
 
 examples = [
     [
-        torch.float16,
         "A storefront with 'Text to Image' written on it.",
         "Expand the following prompt to add more detail:",
         512,