ruslanmv commited on
Commit
b1d31cc
·
1 Parent(s): 9f35428

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -6
app.py CHANGED
@@ -1,17 +1,24 @@
1
- #app.py
2
  import os
 
3
  from flux_app.frontend import Frontend
4
  from flux_app.backend import ModelManager
5
 
6
- if __name__ == "__main__":
7
  # Get the Hugging Face token from an environment variable
8
  hf_token = os.environ.get("HF_TOKEN")
9
-
10
  if not hf_token:
11
- raise ValueError("Hugging Face token (HF_TOKEN) not found in environment variables. Please set it.")
 
12
  model_manager = ModelManager(hf_token=hf_token)
13
-
14
  frontend = Frontend(model_manager)
15
  app = frontend.create_ui()
 
 
16
  app.queue()
17
- app.launch()
 
 
 
 
 
 
1
+ # app.py
2
  import os
3
+
4
  from flux_app.frontend import Frontend
5
  from flux_app.backend import ModelManager
6
 
7
+ def main():
8
  # Get the Hugging Face token from an environment variable
9
  hf_token = os.environ.get("HF_TOKEN")
 
10
  if not hf_token:
11
+ raise ValueError("Hugging Face token (HF_TOKEN) not found in environment variables. Please set it.")
12
+
13
  model_manager = ModelManager(hf_token=hf_token)
 
14
  frontend = Frontend(model_manager)
15
  app = frontend.create_ui()
16
+
17
+ # Enable request queuing. (No extra keyword arguments here.)
18
  app.queue()
19
+
20
+ # Launch the app with the specified server configuration.
21
+ app.launch(server_name="0.0.0.0", server_port=7860, share=False)
22
+
23
+ if __name__ == "__main__":
24
+ main()