Update app.py

app.py

@@ -66,11 +66,6 @@ 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",
@@ -99,24 +94,19 @@ if uploaded_files:
             inputs = inputs.to(device)  # Ensure inputs are on the same device as the model
 
             # Inference: Generation of the output
-            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
+            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])
 
             # Clear memory after processing each file
             del image, inputs, generated_ids, generated_ids_trimmed, output_text