Vtr100 commited on
Commit
63173b0
·
verified ·
1 Parent(s): 0c6ecd1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -8,13 +8,14 @@ MODEL_PATH = os.getenv("MODEL_PATH", "baseline_pipe.pkl")
8
  baseline = None
9
  if os.path.exists(MODEL_PATH):
10
  baseline = joblib.load(MODEL_PATH)
 
 
11
 
12
  def predict_sentiment(text: str):
13
  if not text or text.strip() == "":
14
  return {"predição": "n/a", "confiança": 0.0}
15
  if baseline is None:
16
- # Mensagem amigável quando o modelo ainda não foi enviado
17
- return {"erro": "Modelo não encontrado. Envie baseline_pipe.pkl nos Files do Space."}
18
  proba = baseline.predict_proba([text])[0]
19
  pred = int(np.argmax(proba))
20
  label = "positivo" if pred == 1 else "negativo"
@@ -26,8 +27,8 @@ demo = gr.Interface(
26
  inputs=gr.Textbox(label="Digite uma avaliação de produto"),
27
  outputs=gr.JSON(label="Resultado"),
28
  title="Análise de Sentimentos (Amazon Polarity)",
29
- description="Envie o arquivo baseline_pipe.pkl na aba Files para ativar o modelo."
30
  )
31
 
32
  if __name__ == "__main__":
33
- demo.launch()
 
8
  baseline = None
9
  if os.path.exists(MODEL_PATH):
10
  baseline = joblib.load(MODEL_PATH)
11
+ else:
12
+ print(f"Aviso: modelo '{MODEL_PATH}' não encontrado. A predição usará modo simulado.")
13
 
14
  def predict_sentiment(text: str):
15
  if not text or text.strip() == "":
16
  return {"predição": "n/a", "confiança": 0.0}
17
  if baseline is None:
18
+ return {"predição": "modelo não carregado", "confiança": 0.0}
 
19
  proba = baseline.predict_proba([text])[0]
20
  pred = int(np.argmax(proba))
21
  label = "positivo" if pred == 1 else "negativo"
 
27
  inputs=gr.Textbox(label="Digite uma avaliação de produto"),
28
  outputs=gr.JSON(label="Resultado"),
29
  title="Análise de Sentimentos (Amazon Polarity)",
30
+ description="Pipeline TF-IDF + Regressão Logística. Faça upload de baseline_pipe.pkl gerado no notebook."
31
  )
32
 
33
  if __name__ == "__main__":
34
+ demo.launch()