Update gradio_app.py

gradio_app.py

@@ -1,50 +1,50 @@
-import gradio as gr
-import requests
-
-BASE_URL = "https://citegraph-1099254244451.us-east1.run.app/"
-HEADERS = {"Content-Type": "application/json"}
-
-
-def load_documents(file_path):
-    print(file_path)
-    url = f"{BASE_URL}/api/predict"
-    if file_path is not None:
-        with open(file_path.name, "rb") as f:
-            file = {"file": f}
-            response = requests.post(url, files=file)
-        f.close()
-
-    output = response.json()
-    print(output)
-    return f"## Output: \n# <div align='center'>{output['predicted_label']}</div>"
-
-
-def createUI():
-    css = """
-        h1 {
-            text-align: center;
-            display:block;
-        }
-    """
-    with gr.Blocks(css=css) as demo:
-        with gr.Row():
-            with gr.Column(scale=1):
-                # pdf_display = gr.Image(
-                #     label="Uploaded PDF Page", interactive=False, height=680
-                # )
-                upload_btn = gr.File(label="Upload a PDF", file_types=[".pdf"])
-
-        with gr.Row():
-            class_label = gr.Markdown("## Output: ")
-
-        upload_btn.upload(
-            fn=load_documents,
-            inputs=[upload_btn],
-            outputs=[class_label],
-        )
-    return demo
-
-
-if __name__ == "__main__":
-    demo = createUI()
-    demo.launch(pwa=True, share=False)
+import gradio as gr
+import requests
+
+BASE_URL = "https://citegraph-1099254244451.us-east1.run.app/"
+HEADERS = {"Content-Type": "application/json"}
+
+
+def load_documents(file_path):
+    print(file_path)
+    url = f"{BASE_URL}/api/predict"
+    if file_path is not None:
+        with open(file_path.name, "rb") as f:
+            file = {"file": f}
+            response = requests.post(url, files=file)
+        f.close()
+
+    output = response.json()
+    print(output)
+    return f"## <div align='center'>Output: </div>\n# <div align='center'>{output['predicted_label']}</div>"
+
+
+def createUI():
+    css = """
+        h1 {
+            text-align: center;
+            display:block;
+        }
+    """
+    with gr.Blocks(css=css) as demo:
+        with gr.Row():
+            with gr.Column(scale=1):
+                # pdf_display = gr.Image(
+                #     label="Uploaded PDF Page", interactive=False, height=680
+                # )
+                upload_btn = gr.File(label="Upload a PDF", file_types=[".pdf"])
+
+        with gr.Row():
+            class_label = gr.Markdown("## Output: ")
+
+        upload_btn.upload(
+            fn=load_documents,
+            inputs=[upload_btn],
+            outputs=[class_label],
+        )
+    return demo
+
+
+if __name__ == "__main__":
+    demo = createUI()
+    demo.launch(pwa=True, share=False)