gilramos commited on
Commit
be4ba0f
1 Parent(s): 918502d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -11
app.py CHANGED
@@ -24,16 +24,6 @@ model_list = [
24
  "knowhate/HateBERTimbau-yt-tt",
25
  ]
26
 
27
- user_friendly_name = {
28
- "knowhate/HateBERTimbau": "HateBERTimbau (Original)",
29
- "knowhate/HateBERTimbau-youtube": "HateBERTimbau (YouTube)",
30
- "knowhate/HateBERTimbau-twitter": "HateBERTimbau (Twitter)",
31
- "knowhate/HateBERTimbau-yt-tt": "HateBERTimbau (YouTube + Twitter)",
32
- }
33
-
34
- reverse_user_friendly_name = { v:k for k,v in user_friendly_name.items() }
35
- user_friendly_name_list = list(user_friendly_name.values())
36
-
37
  #pipe = pipeline("text-classification", model="knowhate/HateBERTimbau")
38
  #demo = gr.Interface.from_pipeline(pipe)
39
  #demo.launch()
@@ -49,7 +39,7 @@ def predict(text, chosen_model):
49
 
50
  inputs = [
51
  gr.Textbox(label="Text", value= app_examples[0][0]),
52
- gr.Dropdown(label="Model", choices=user_friendly_name_list, value=user_friendly_name_list[2])
53
  ]
54
 
55
  outputs = [
 
24
  "knowhate/HateBERTimbau-yt-tt",
25
  ]
26
 
 
 
 
 
 
 
 
 
 
 
27
  #pipe = pipeline("text-classification", model="knowhate/HateBERTimbau")
28
  #demo = gr.Interface.from_pipeline(pipe)
29
  #demo.launch()
 
39
 
40
  inputs = [
41
  gr.Textbox(label="Text", value= app_examples[0][0]),
42
+ gr.Dropdown(label="Model", choices=model_list, value=model_list[2])
43
  ]
44
 
45
  outputs = [