Updated with OCR model and Gradio integration

app.py

@@ -8,7 +8,9 @@ repo_id = "OpenGVLab/InternVL2-1B"
 tokenizer = AutoTokenizer.from_pretrained(repo_id, trust_remote_code=True)
 processor = AutoProcessor.from_pretrained(repo_id, trust_remote_code=True)
 model = AutoModel.from_pretrained(
-    repo_id, trust_remote_code=True, torch_dtype=torch.float16
+    repo_id,
+    trust_remote_code=True,
+    torch_dtype=torch.float16  # Use half-precision for efficiency
 )
 
 # Move model to the appropriate device
@@ -33,4 +35,4 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=True)