Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
| 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="
|
| 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()
|