lorenzoscottb commited on
Commit
5d70280
β€’
1 Parent(s): 2736ee9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -18
app.py CHANGED
@@ -78,23 +78,25 @@ interface_words = gr.Interface(
78
  examples=example_main,
79
  )
80
 
81
- interface_model_L = gr.Interface.load(
82
- "models/DReAMy-lib/xlm-roberta-large-DreamBank-emotion-presence",
83
  examples=examples,
84
  title="SA Large Multilingual",
85
  )
 
86
 
87
- interface_model_S = gr.Interface.load(
88
- "models/DReAMy-lib/bert-base-cased-DreamBank-emotion-presence",
89
- examples=examples[0],
90
- title="SA Base English-Only",
91
- )
92
 
93
- interface_model_G = gr.Interface.load(
94
- "models/DReAMy-lib/t5-base-DreamBank-Generation-Emot-Char",
95
- examples=examples_g,
96
- title="SA Generation",
97
- )
 
 
 
 
 
 
98
 
99
  # interface_model_RE = gr.Interface(
100
  # text_to_graph,
@@ -110,13 +112,13 @@ interface_model_G = gr.Interface.load(
110
  # cache_examples=True,
111
  # )
112
 
113
- interface_model_NER = gr.Interface.load(
114
- "models/DReAMy-lib/t5-base-DreamBank-Generation-NER-Char",
115
- examples=examples_g,
116
- title="NER Generation",
117
  )
118
 
119
  gr.TabbedInterface(
120
- [interface_words, interface_model_NER, interface_model_L, interface_model_S, interface_model_G],
121
- ["Main", "NER Generation", "SA Large Multilingual", "SA Base En", "SA En Generation"],
122
  ).launch()
 
78
  examples=example_main,
79
  )
80
 
81
+ interface_model_L = gr.Interface(
82
+ description=description_L,
83
  examples=examples,
84
  title="SA Large Multilingual",
85
  )
86
+ interface_model_L.load("models/DReAMy-lib/xlm-roberta-large-DreamBank-emotion-presence")
87
 
 
 
 
 
 
88
 
89
+ # interface_model_S = gr.Interface.load(
90
+ # "models/DReAMy-lib/bert-base-cased-DreamBank-emotion-presence",
91
+ # examples=examples[0],
92
+ # title="SA Base English-Only",
93
+ # )
94
+
95
+ # interface_model_G = gr.Interface.load(
96
+ # "models/DReAMy-lib/t5-base-DreamBank-Generation-Emot-Char",
97
+ # examples=examples_g,
98
+ # title="SA Generation",
99
+ # )
100
 
101
  # interface_model_RE = gr.Interface(
102
  # text_to_graph,
 
112
  # cache_examples=True,
113
  # )
114
 
115
+ # interface_model_NER = gr.Interface.load(
116
+ # "models/DReAMy-lib/t5-base-DreamBank-Generation-NER-Char",
117
+ # examples=examples_g,
118
+ # title="NER Generation",
119
  )
120
 
121
  gr.TabbedInterface(
122
+ [interface_words, interface_model_L],
123
+ ["Main","SA Large Multilingual"],
124
  ).launch()