IBKcode commited on
Commit
3036a2a
·
verified ·
1 Parent(s): 28d6a85

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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-small",
11
  device=-1)
12
  emotion_model = pipeline(
13
  "text-classification",
14
- model="Panda0116/emotion-classification-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
  )