ProfRod100 commited on
Commit
d64b8ba
·
verified ·
1 Parent(s): 9398817

Upload app (1).py

Browse files
Files changed (1) hide show
  1. app (1).py +33 -0
app (1).py ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ import gradio as gr
3
+ import joblib
4
+ import numpy as np
5
+ import os
6
+
7
+ # Carrega o pipeline baseline (TF-IDF + LogisticRegression)
8
+ MODEL_PATH = os.getenv("MODEL_PATH", "baseline_pipe.pkl")
9
+
10
+ if not os.path.exists(MODEL_PATH):
11
+ raise FileNotFoundError(f"Arquivo de modelo não encontrado: {MODEL_PATH}. Faça upload de baseline_pipe.pkl.")
12
+
13
+ baseline = joblib.load(MODEL_PATH)
14
+
15
+ def predict_sentiment(text: str):
16
+ if not text or text.strip() == "":
17
+ return {"predição": "n/a", "confiança": 0.0}
18
+ proba = baseline.predict_proba([text])[0]
19
+ pred = int(np.argmax(proba))
20
+ label = "positivo" if pred == 1 else "negativo"
21
+ conf = float(np.max(proba))
22
+ return {"predição": label, "confiança": conf}
23
+
24
+ demo = gr.Interface(
25
+ fn=predict_sentiment,
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="Pipeline TF-IDF + Regressão Logística. Faça upload de baseline_pipe.pkl gerado no notebook."
30
+ )
31
+
32
+ if __name__ == "__main__":
33
+ demo.launch()