Hjgugugjhuhjggg commited on
Commit
f7ba4cc
verified
1 Parent(s): daa26d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -10
app.py CHANGED
@@ -70,6 +70,7 @@ def liberar_recursos():
70
  os.kill(proc.info['pid'], 9)
71
  time.sleep(60)
72
 
 
73
  threading.Thread(target=keep_alive, daemon=True).start()
74
  threading.Thread(target=liberar_recursos, daemon=True).start()
75
 
@@ -114,22 +115,22 @@ async def resumen(texto: str):
114
 
115
  @app.post("/similitud")
116
  async def similitud(texto1: str, texto2: str):
117
- try:
118
- texto1_vec = vectorizer.fit_transform([texto1])
119
- texto2_vec = vectorizer.transform([texto2])
120
- similitud = cosine_similarity(texto1_vec, texto2_vec)
121
- return {"similitud": similitud[0][0]}
122
- except Exception as e:
123
- logging.error(f"Error en similitud: {e}")
124
- return {"error": "Error en similitud"}
125
 
126
  @app.get("/modelos")
127
  async def modelos():
128
- return {"modelos": list(cadenas.keys())}
129
 
130
  @app.get("/estado")
131
  async def estado():
132
- return {"estado": "activo"}
133
 
134
  if __name__ == "__main__":
135
  uvicorn.run(app, host="0.0.0.0", port=8000)
 
70
  os.kill(proc.info['pid'], 9)
71
  time.sleep(60)
72
 
73
+ # Crear hilos para mantener la aplicaci贸n viva y liberar recursos
74
  threading.Thread(target=keep_alive, daemon=True).start()
75
  threading.Thread(target=liberar_recursos, daemon=True).start()
76
 
 
115
 
116
  @app.post("/similitud")
117
  async def similitud(texto1: str, texto2: str):
118
+ try:
119
+ texto1_vec = vectorizer.fit_transform([texto1])
120
+ texto2_vec = vectorizer.transform([texto2])
121
+ similitud = cosine_similarity(texto1_vec, texto2_vec)
122
+ return {"similitud": similitud[0][0]}
123
+ except Exception as e:
124
+ logging.error(f"Error en similitud: {e}")
125
+ return {"error": "Error en similitud"}
126
 
127
  @app.get("/modelos")
128
  async def modelos():
129
+ return {"modelos": list(cadenas.keys())}
130
 
131
  @app.get("/estado")
132
  async def estado():
133
+ return {"estado": "activo"}
134
 
135
  if __name__ == "__main__":
136
  uvicorn.run(app, host="0.0.0.0", port=8000)