Spaces:
Running
on
Zero
Running
on
Zero
lucapinello
commited on
Commit
·
0b617df
1
Parent(s):
b074e28
update
Browse files- app.py +14 -2
- requirements.txt +5 -1
app.py
CHANGED
@@ -238,14 +238,26 @@ if __name__ == "__main__":
|
|
238 |
parser = argparse.ArgumentParser(description="DNA-Diffusion Gradio App")
|
239 |
parser.add_argument("--share", action="store_true", help="Create a public shareable link")
|
240 |
parser.add_argument("--port", type=int, default=7860, help="Port to run the app on")
|
241 |
-
parser.add_argument("--host", type=str, default="
|
242 |
args = parser.parse_args()
|
243 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
244 |
logger.info(f"Starting DNA-Diffusion Gradio app on {args.host}:{args.port}")
|
245 |
|
246 |
demo.launch(
|
247 |
share=args.share,
|
248 |
server_name=args.host,
|
249 |
server_port=args.port,
|
250 |
-
inbrowser=
|
|
|
251 |
)
|
|
|
238 |
parser = argparse.ArgumentParser(description="DNA-Diffusion Gradio App")
|
239 |
parser.add_argument("--share", action="store_true", help="Create a public shareable link")
|
240 |
parser.add_argument("--port", type=int, default=7860, help="Port to run the app on")
|
241 |
+
parser.add_argument("--host", type=str, default="0.0.0.0", help="Host to run the app on")
|
242 |
args = parser.parse_args()
|
243 |
|
244 |
+
# For Hugging Face Spaces deployment
|
245 |
+
import os
|
246 |
+
if os.getenv("SPACE_ID"):
|
247 |
+
# Running on Hugging Face Spaces
|
248 |
+
args.host = "0.0.0.0"
|
249 |
+
args.port = 7860
|
250 |
+
args.share = False
|
251 |
+
inbrowser = False
|
252 |
+
else:
|
253 |
+
inbrowser = True
|
254 |
+
|
255 |
logger.info(f"Starting DNA-Diffusion Gradio app on {args.host}:{args.port}")
|
256 |
|
257 |
demo.launch(
|
258 |
share=args.share,
|
259 |
server_name=args.host,
|
260 |
server_port=args.port,
|
261 |
+
inbrowser=inbrowser,
|
262 |
+
ssr_mode=False # Disable SSR for better compatibility
|
263 |
)
|
requirements.txt
CHANGED
@@ -11,4 +11,8 @@ accelerate>=0.20.0
|
|
11 |
|
12 |
# Note: DNA-Diffusion itself should be installed separately using uv:
|
13 |
# git clone https://github.com/pinellolab/DNA-Diffusion.git
|
14 |
-
# cd DNA-Diffusion && uv sync
|
|
|
|
|
|
|
|
|
|
11 |
|
12 |
# Note: DNA-Diffusion itself should be installed separately using uv:
|
13 |
# git clone https://github.com/pinellolab/DNA-Diffusion.git
|
14 |
+
# cd DNA-Diffusion && uv sync
|
15 |
+
|
16 |
+
# For Hugging Face Spaces, you may need to add:
|
17 |
+
# --extra-index-url https://download.pytorch.org/whl/cu118
|
18 |
+
# And ensure GPU is enabled in Space settings
|