Update app.py

app.py

@@ -66,6 +66,11 @@ if uploaded_files:
                 st.error("Unsupported file type.")
                 continue
 
+            # Ensure the image is loaded correctly
+            if image is None:
+                st.error("Failed to load the image.")
+                continue
+
             messages = [
                 {
                     "role": "user",
@@ -94,19 +99,24 @@ if uploaded_files:
             inputs = inputs.to(device)  # Ensure inputs are on the same device as the model
 
             # Inference: Generation of the output
-            generated_ids = model.generate(**inputs, max_new_tokens=512)
-            generated_ids_trimmed = [
-                out_ids[len(in_ids) :] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)
-            ]
-            output_text = processor.batch_decode(
-                generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
-            )
-
-            st.write("Description:")
-            st.write(output_text[0])
-
-            # Append the output text to the list
-            all_output_texts.append(output_text[0])
+            try:
+                generated_ids = model.generate(**inputs, max_new_tokens=512)
+                generated_ids_trimmed = [
+                    out_ids[len(in_ids) :] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)
+                ]
+                output_text = processor.batch_decode(
+                    generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
+                )
+
+                st.write("Description:")
+                st.write(output_text[0])
+
+                # Append the output text to the list
+                all_output_texts.append(output_text[0])
+
+            except Exception as e:
+                st.error(f"Error during generation: {e}")
+                continue
 
             # Clear memory after processing each file
             del image, inputs, generated_ids, generated_ids_trimmed, output_text