Updated with OCR model and Gradio integration

app.py

@@ -1,13 +1,12 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoProcessor, AutoConfig, AutoModelForSeq2SeqLM
+from transformers import AutoTokenizer, AutoProcessor, AutoModel
 
 repo_id = "OpenGVLab/InternVL2-1B"
 
-# Load the tokenizer, processor, and model directly from the Hugging Face Hub
+# Load the tokenizer, processor, and model directly from the Hub
 tokenizer = AutoTokenizer.from_pretrained(repo_id, trust_remote_code=True)
 processor = AutoProcessor.from_pretrained(repo_id, trust_remote_code=True)
-config = AutoConfig.from_pretrained(repo_id, trust_remote_code=True)
-model = AutoModelForSeq2SeqLM.from_pretrained(repo_id, config=config, trust_remote_code=True)
+model = AutoModel.from_pretrained(repo_id, trust_remote_code=True)
 
 def analyze_image(image):
     img = image.convert("RGB")

requirements.txt

@@ -1,6 +1,4 @@
 transformers>=4.31.0
 gradio>=3.35.2
 torch>=1.9.0
-huggingface_hub>=0.14.1
-pillow
-accelerate
+pillow