Update app.py

app.py

@@ -30,7 +30,7 @@ def generate(your_prompt, max_new_tokens, repetition_penalty, temperature, model
     model.to(dtype)
     
     input_text = f"Expand the following prompt to add more detail: {your_prompt}"
-    input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to(device).to(dtype)
+    input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to(device)
         
     outputs = model.generate(
         input_ids,