Handling uploaded file object

custom_got/modeling_GOT.py

@@ -469,7 +469,7 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
             config.im_start_token, config.im_end_token = 151857, 151858
 
     def load_image(self, image_file):
-        if image_file.startswith('http') or image_file.startswith('https'):
+        if isinstance(image_file, str) and (image_file.startswith('http') or image_file.startswith('https')):
             response = requests.get(image_file)
             image = Image.open(BytesIO(response.content)).convert('RGB')
         else:

main_got.py

@@ -22,5 +22,5 @@ def extract_text(image_path):
 
     image_file = image_path
     # Extract text
-    res = model.chat(tokenizer, image_file, ocr_type='ocr', gradio_input=True)
+    res = model.chat(tokenizer, image_file, ocr_type='ocr')
     return res