Update app.py

app.py

@@ -57,7 +57,7 @@ pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 # pipe.to(device)
-model2.to(device)
+
 model.to(device)
 pipe.to(device)
 
@@ -110,7 +110,7 @@ def generate_caption_and_image(image, f, p, d):
         # return generated_image, generated_image1
         messages = [{"role": "user", "content": [{"type": "image"},{"type": "text", "text": "If I had to write a haiku for this one, it would be: "}]}]
         input_text = processor.apply_chat_template(messages, add_generation_prompt=True)
-        inputs = processor(image,input_text,add_special_tokens=False,return_tensors="pt").to(model.device)
+        inputs = processor(image,input_text,add_special_tokens=False,return_tensors="pt").to(device)
         
         output = model.generate(**inputs, max_new_tokens=30)
         caption =processor.decode(output[0])