TayebBou commited on
Commit
97e91c0
·
verified ·
1 Parent(s): fc27bb3

Déploiement automatique depuis GitHub Actions 🚀

Browse files
Files changed (1) hide show
  1. app.py +3 -5
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
- # On est sur HF Space → juste lancer Gradio
147
  if __name__ == "__main__":
148
  demo.launch()
149
  else:
150
- # En local ou autre → FastAPI + Gradio montés ensemble
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)