Update app.py

app.py

@@ -1,21 +1,32 @@
 import os
-os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "0"  # disable fast download before importing transformers
+os.environ.setdefault("HF_HUB_ENABLE_HF_TRANSFER", "1")  # optional speed-up
 
 import gradio as gr
+import torch
 from transformers import pipeline
 
-ocr_pipeline = pipeline("image-to-text", model="deepseek-ai/DeepSeek-OCR")
+# important: trust_remote_code=True avoids the interactive prompt
+dtype = torch.float16 if torch.cuda.is_available() else torch.float32
+
+ocr = pipeline(
+    "image-to-text",
+    model="deepseek-ai/DeepSeek-OCR",
+    trust_remote_code=True,
+    device_map="auto",          # uses GPU if available, else CPU
+    torch_dtype=dtype,
+)
 
 def ocr_image(image):
-    result = ocr_pipeline(image)
-    return result[0]["generated_text"]
+    out = ocr(image)
+    # pipeline returns a list of dicts like [{'generated_text': '...'}]
+    return out[0]["generated_text"] if out else ""
 
 demo = gr.Interface(
     fn=ocr_image,
     inputs=gr.Image(type="pil"),
     outputs="text",
     title="DeepSeek OCR",
-    description="Upload an image and get extracted text using DeepSeek-OCR.",
+    description="Upload an image and get extracted text using DeepSeek-OCR."
 )
 
 if __name__ == "__main__":