Update app.py

app.py

@@ -19,11 +19,11 @@ 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)
+    _image.save(image_path)  # Save the image using the underscore variable
     st.write(f"Saved image to {image_path}")
 
     try:
@@ -72,7 +72,7 @@ if uploaded_image:
                 tokenizer, model = load_model(model_option)
                 
                 # Run OCR and cache the result using @st.cache_data
-                result_text = run_GOT(image, tokenizer, model)
+                result_text = run_GOT(image, tokenizer, model)  # Pass the original image here
                 
                 if "Error" not in result_text:
                     # Keyword input for search