Update app.py
Browse files
app.py
CHANGED
|
@@ -7,11 +7,11 @@ import gradio as gr
|
|
| 7 |
# --- MODEL LOADING ---
|
| 8 |
text_model = pipeline(
|
| 9 |
"text2text-generation",
|
| 10 |
-
model="google/flan-t5-
|
| 11 |
device=-1)
|
| 12 |
emotion_model = pipeline(
|
| 13 |
"text-classification",
|
| 14 |
-
model="
|
| 15 |
device=-1,
|
| 16 |
return_all_scores=True
|
| 17 |
)
|
|
|
|
| 7 |
# --- MODEL LOADING ---
|
| 8 |
text_model = pipeline(
|
| 9 |
"text2text-generation",
|
| 10 |
+
model="google/flan-t5-base",
|
| 11 |
device=-1)
|
| 12 |
emotion_model = pipeline(
|
| 13 |
"text-classification",
|
| 14 |
+
model="SamLowe/roberta-base-go_emotions",
|
| 15 |
device=-1,
|
| 16 |
return_all_scores=True
|
| 17 |
)
|