Upload app.py

app.py

@@ -0,0 +1,39 @@
+import gradio as gr
+import requests
+from io import BytesIO
+import os
+
+def call_predict_api(image):
+    # Convert PIL images to base64 string
+    src_buffer = BytesIO()
+    image.save(src_buffer, format='PNG')
+    src_buffer.seek(0)
+    
+    # Prepare files for upload
+    files = {
+        "image": ("src_image.png", src_buffer, "image/png"),
+    }
+    headers = {"X-API-Key": os.environ["api_key"]}
+    response = requests.post(os.environ["endpoint"], files=files, headers=headers)
+    if response.status_code != 200:
+        raise Exception(f"API Error: {response.text}")
+    result = response.json()["result"]
+    return result.replace("\n", "\n\n")
+
+if __name__ == "__main__":
+    iface = gr.Interface(
+        fn=call_predict_api,
+        inputs=[
+            gr.Image(
+                    sources=["upload"],
+                    type="pil",
+                    label="Garment Image",
+                    width=512,
+                    height=512,
+                ),
+        ],
+        outputs=[gr.Markdown(label="Result")],
+        title="BBC-OCR",
+        description="A simple OCR tool to extract text from images.",
+    )
+    iface.launch(auth=[(os.environ["username"], os.environ["password"])], show_api=False)