Spaces:
Running
Running
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
|