Spaces:
Sleeping
Sleeping
Commit ·
c87585c
1
Parent(s): 6e7cdea
commit 00000002
Browse files
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 |
-
|
| 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)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|