lucapinello commited on
Commit
0b617df
·
1 Parent(s): b074e28
Files changed (2) hide show
  1. app.py +14 -2
  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="127.0.0.1", help="Host to run the app on")
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=True
 
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