Update app.py

app.py

@@ -13,19 +13,22 @@ model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True,
 model = model.eval().to(device)
 
 def extract_text(image):
-    # Save the image to a temporary file
+    if image is None:
+        return "No image uploaded", ""
+
     image_path = 'temp_image.png'
     image.save(image_path)
-    # Use the GOT model to extract text
+
     try:
         res = model.chat(tokenizer, image_path, ocr_type='ocr')
-        return res, res  # Return the extracted text and also set it in the state variable
+        return res, res
     except Exception as e:
         return f"Error: {str(e)}", ""
     finally:
         if os.path.exists(image_path):
             os.remove(image_path)
 
+
 def keyword_search(extracted_text, keyword):
     if not extracted_text:
         return "No text extracted yet."