Lord-Raven commited on
Commit
0686401
·
1 Parent(s): eb893c3

Playing with different models.

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -10,11 +10,12 @@ def zero_shot_classification(data_string):
10
  data = json.loads(data_string)
11
  print(data)
12
  results = classifier(data['sequence'], candidate_labels=data['candidate_labels'], hypothesis_template=data['hypothesis_template'], multi_label=data['multi_label'])
13
- return json.dumps(results)
 
14
 
15
  gradio_interface = gradio.Interface(
16
  fn = zero_shot_classification,
17
  inputs = gradio.Textbox(label="JSON Input"),
18
- outputs = gradio.JSON()
19
  )
20
  gradio_interface.launch()
 
10
  data = json.loads(data_string)
11
  print(data)
12
  results = classifier(data['sequence'], candidate_labels=data['candidate_labels'], hypothesis_template=data['hypothesis_template'], multi_label=data['multi_label'])
13
+ response_string = json.dumps(results)
14
+ return response_string
15
 
16
  gradio_interface = gradio.Interface(
17
  fn = zero_shot_classification,
18
  inputs = gradio.Textbox(label="JSON Input"),
19
+ outputs = gradio.Textbox()
20
  )
21
  gradio_interface.launch()