Create app.py

app.py

@@ -0,0 +1,21 @@
+import gradio as gr
+from transformers import pipeline
+
+# This line downloads the model from Hugging Face
+pipe = pipeline("image-text-to-text", model="ibm-granite/granite-docling-258M")
+
+def process_image(image):
+    # This runs the model on the uploaded image
+    result = pipe(image)
+    # This returns the text extracted from the document
+    return str(result)
+
+# This creates the user interface
+demo = gr.Interface(
+    fn=process_image,
+    inputs=gr.Image(type="pil"),
+    outputs="text"
+)
+
+# This launches the app
+demo.launch()