try gradio

Dockerfile

@@ -1,7 +1,7 @@
 # read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
 # you will also find guides on how best to write your Dockerfile
 
-FROM python:3.9
+FROM kevinkune/sbdd:latest
 
 RUN useradd -m -u 1000 user
 

app.py

@@ -1,7 +1,19 @@
-from fastapi import FastAPI
+# from fastapi import FastAPI
 
-app = FastAPI()
+# app = FastAPI()
 
-@app.get("/")
-def greet_json():
-    return {"Hello": "World!"}
+# @app.get("/")
+# def greet_json():
+#     return {"Hello": "World!"}
+
+
+import gradio as gr
+
+from gradio_molecule3d import Molecule3D
+
+with gr.Blocks() as demo:
+    gr.Markdown("# MolCRAFT: Structure-Based Drug Design in Continuous Parameter Space [ICML 2024]")
+
+
+if __name__ == '__main__':
+    demo.launch(share=True)