Calamar49 commited on
Commit
bab73df
·
1 Parent(s): 358074b

final-final

Browse files
Files changed (2) hide show
  1. app.py +3 -4
  2. requirements.txt +1 -1
app.py CHANGED
@@ -17,9 +17,8 @@ device = torch.device("cpu")
17
  MODEL_PATH = "model/best_vision_ensemble_model.pth"
18
  NUM_CLASSES = 156
19
 
20
- # No necesitamos el mapa de etiquetas en el servidor, ya que la app lo manejará
21
- # with open(LABELS_PATH) as f:
22
- # labels_map = json.load(f)
23
 
24
  model = VisionEnsembleModel(num_classes=NUM_CLASSES)
25
  model.load_state_dict(torch.load(MODEL_PATH, map_location=device))
@@ -38,7 +37,7 @@ app = FastAPI(title="API de Clasificación de Orquídeas")
38
 
39
  @app.get("/")
40
  def read_root():
41
- return {"status": "ok", "message": "API de Orquídeas funcionando. Usa el endpoint /predict_for_mobile"}
42
 
43
  # --- 4. Endpoint de API para la App Móvil (devuelve IDs) ---
44
  @app.post("/predict_for_mobile")
 
17
  MODEL_PATH = "model/best_vision_ensemble_model.pth"
18
  NUM_CLASSES = 156
19
 
20
+ # El mapa de etiquetas ya no es necesario en el servidor
21
+ # ya que la app móvil se encargará de la traducción.
 
22
 
23
  model = VisionEnsembleModel(num_classes=NUM_CLASSES)
24
  model.load_state_dict(torch.load(MODEL_PATH, map_location=device))
 
37
 
38
  @app.get("/")
39
  def read_root():
40
+ return {"status": "ok", "message": "API de Orquídeas funcionando. Usa el endpoint POST en /predict_for_mobile"}
41
 
42
  # --- 4. Endpoint de API para la App Móvil (devuelve IDs) ---
43
  @app.post("/predict_for_mobile")
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- # requirements.txt
2
  fastapi
3
  uvicorn[standard]
4
  python-multipart
 
1
+ # requirements.txt (versión final solo para API)
2
  fastapi
3
  uvicorn[standard]
4
  python-multipart