FrederickSundeep commited on
Commit
c87585c
·
1 Parent(s): 6e7cdea

commit 00000002

Browse files
Files changed (1) hide show
  1. app.py +19 -6
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import os
2
  import time
 
3
  import torch
4
  import gradio as gr
5
  from flask import Flask, request, Response
@@ -161,10 +162,22 @@ with gr.Blocks() as demo:
161
  clear.click(lambda: None, None, chatbot, queue=False)
162
 
163
  # ✅ Run Gradio when in HF Spaces, else Flask for local dev
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
164
  if __name__ == "__main__":
165
- if os.environ.get("SPACE_BUILD", "false").lower() == "true":
166
- demo.launch(server_name="0.0.0.0", server_port=7860)
167
- else:
168
- print("🔧 Warming up...")
169
- _ = generate_full_reply("Hello", [])
170
- app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))
 
1
  import os
2
  import time
3
+ import threading
4
  import torch
5
  import gradio as gr
6
  from flask import Flask, request, Response
 
162
  clear.click(lambda: None, None, chatbot, queue=False)
163
 
164
  # ✅ Run Gradio when in HF Spaces, else Flask for local dev
165
+ # if __name__ == "__main__":
166
+ # if os.environ.get("SPACE_BUILD", "false").lower() == "true":
167
+ # demo.launch(server_name="0.0.0.0", server_port=7860)
168
+ # else:
169
+ # print("🔧 Warming up...")
170
+ # _ = generate_full_reply("Hello", [])
171
+ # app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))
172
+
173
+
174
+ # ---------------- Run both ----------------
175
+ def run_flask():
176
+ app.run(host="0.0.0.0", port=8000)
177
+
178
+ # Start Flask in a background thread
179
+ threading.Thread(target=run_flask, daemon=True).start()
180
+
181
+ # Gradio runs on port 7860 in HF Spaces
182
  if __name__ == "__main__":
183
+ demo.launch(server_name="0.0.0.0", server_port=7860)