Spaces:
Running
Running
Déploiement automatique depuis GitHub Actions 🚀
Browse files
app.py
CHANGED
|
@@ -136,21 +136,19 @@ def predict_api(item: dict):
|
|
| 136 |
probs_head_random = predict_proba_from_model(mdl_head_random, text)
|
| 137 |
return {"fine_tuned": probs_ft, "head_random": probs_head_random}
|
| 138 |
|
| 139 |
-
# for k, v in os.environ.items():
|
| 140 |
-
# print(f"{k}={v}")
|
| 141 |
|
| 142 |
# --- Mount Gradio in HF Space friendly way ---
|
| 143 |
IS_HF_SPACE = os.getenv("SYSTEM") == "spaces"
|
| 144 |
|
| 145 |
if IS_HF_SPACE:
|
| 146 |
-
#
|
| 147 |
if __name__ == "__main__":
|
| 148 |
demo.launch()
|
| 149 |
else:
|
| 150 |
-
#
|
| 151 |
mount_gradio_app(app, demo, path="/")
|
| 152 |
|
| 153 |
if __name__ == "__main__":
|
| 154 |
import uvicorn
|
| 155 |
|
| 156 |
-
uvicorn.run("app:app", host="0.0.0.0", port=7860, reload=True)
|
|
|
|
| 136 |
probs_head_random = predict_proba_from_model(mdl_head_random, text)
|
| 137 |
return {"fine_tuned": probs_ft, "head_random": probs_head_random}
|
| 138 |
|
|
|
|
|
|
|
| 139 |
|
| 140 |
# --- Mount Gradio in HF Space friendly way ---
|
| 141 |
IS_HF_SPACE = os.getenv("SYSTEM") == "spaces"
|
| 142 |
|
| 143 |
if IS_HF_SPACE:
|
| 144 |
+
# In HF Space → launch Gradio only
|
| 145 |
if __name__ == "__main__":
|
| 146 |
demo.launch()
|
| 147 |
else:
|
| 148 |
+
# In local or else → FastAPI + Gradio mounted together
|
| 149 |
mount_gradio_app(app, demo, path="/")
|
| 150 |
|
| 151 |
if __name__ == "__main__":
|
| 152 |
import uvicorn
|
| 153 |
|
| 154 |
+
uvicorn.run("app:app", host="0.0.0.0", port=7860, reload=True)
|