Update app.py

app.py

@@ -16,19 +16,23 @@ uploaded_image = st.file_uploader("Upload an image for OCR", type=["jpg", "jpeg"
 
 if uploaded_image is not None:
     # Open and display the uploaded image
-    image = Image.open(uploaded_image)
+    image = Image.open(uploaded_image).convert("RGB")  # Ensure image is in RGB format
     st.image(image, caption="Uploaded Image", use_column_width=True)
 
-    # Convert image to suitable format
-    inputs = processor(images=image, return_tensors="pt")
-
-    # Perform OCR
-    with torch.no_grad():
-        outputs = model.generate(**inputs)
-
-    # Decode the generated text
-    text = processor.decode(outputs[0], skip_special_tokens=True)
-
-    # Display the OCR result
-    st.write("Extracted Text:")
-    st.text(text)
+    # Convert image to a suitable format and ensure it's a batch (list of images)
+    try:
+        # Convert image to the right format for the processor
+        inputs = processor(images=[image], return_tensors="pt")  # Put image in a list
+        
+        # Perform OCR
+        with torch.no_grad():
+            outputs = model.generate(**inputs)
+
+        # Decode the generated text
+        text = processor.decode(outputs[0], skip_special_tokens=True)
+        
+        # Display the OCR result
+        st.write("Extracted Text:")
+        st.text(text)
+    except Exception as e:
+        st.error(f"An error occurred: {str(e)}")