joseluis12 commited on
Commit
84c40c4
verified
1 Parent(s): 8d26471

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -1,7 +1,11 @@
 
1
  from fastapi import FastAPI
2
  from pydantic import BaseModel
3
  from transformers import pipeline
4
 
 
 
 
5
  # Inicializa la aplicaci贸n FastAPI
6
  app = FastAPI()
7
 
@@ -44,7 +48,7 @@ async def procesar_oferta(oferta: JobOffer):
44
  elif etiqueta == "B-ORG":
45
  resultado["empresa"] = palabra if not resultado["empresa"] else f"{resultado['empresa']} {palabra}"
46
  elif etiqueta == "B-LOC":
47
- resultado["ubicaci贸n"] = palabra if not resultado["ubicaci贸n"] else f"{resultado['ubicaci贸n']} {palabra}"
48
  elif etiqueta == "B-SALARY":
49
  resultado["salario"] = palabra if not resultado["salario"] else f"{resultado['salario']} {palabra}"
50
  elif etiqueta == "B-EXP":
 
1
+ import os
2
  from fastapi import FastAPI
3
  from pydantic import BaseModel
4
  from transformers import pipeline
5
 
6
+ # Configurar la cach茅 en un directorio accesible
7
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp"
8
+
9
  # Inicializa la aplicaci贸n FastAPI
10
  app = FastAPI()
11
 
 
48
  elif etiqueta == "B-ORG":
49
  resultado["empresa"] = palabra if not resultado["empresa"] else f"{resultado['empresa']} {palabra}"
50
  elif etiqueta == "B-LOC":
51
+ resultado["ubicacion"] = palabra if not resultado["ubicacion"] else f"{resultado['ubicacion']} {palabra}"
52
  elif etiqueta == "B-SALARY":
53
  resultado["salario"] = palabra if not resultado["salario"] else f"{resultado['salario']} {palabra}"
54
  elif etiqueta == "B-EXP":