Update app.py

app.py

@@ -51,8 +51,7 @@ Below is the simple caption of an image with text. Please deduce the detailed de
         tokenize=False,
         add_generation_prompt=True
     )
-    # model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
-    model_inputs = tokenizer([text], return_tensors="pt")
+    model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
 
     generated_ids = model.generate(
         **model_inputs,