Update app.py

app.py

@@ -19,20 +19,17 @@ def load_model(model_name):
 
 # Function to run the GOT model for multilingual OCR
 @st.cache_data
-def run_GOT(_image, tokenizer, model):
+def run_GOT(_image, _tokenizer, _model):
     unique_id = str(uuid.uuid4())
     image_path = f"{unique_id}.png"
     
     _image.save(image_path)  # Save the image using the underscore variable
-    st.write(f"Saved image to {image_path}")
 
     try:
         # Use the model to extract text
-        res = model.chat(tokenizer, image_path, ocr_type='ocr')  # Extract plain text
-        st.write(f"Raw result: {res}")  # Debug output
+        res = _model.chat(_tokenizer, image_path, ocr_type='ocr')  # Extract plain text
         return res
     except Exception as e:
-        st.error(f"Error: {str(e)}")  # Display any errors
         return f"Error: {str(e)}"
     finally:
         # Clean up the saved image
@@ -85,4 +82,3 @@ if uploaded_image:
                     st.markdown(highlighted_text, unsafe_allow_html=True)
                 else:
                     st.error(result_text)
-