Lord-Raven commited on
Commit
cd2c95d
·
1 Parent(s): b3cb286

Messing with configuration.

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -52,13 +52,11 @@ def classify(data_string, request: gradio.Request):
52
  # if 'task' in data and data['task'] == 'few_shot_classification':
53
  # return few_shot_classification(data)
54
  # else:
55
- return zero_shot_classification(data)
56
 
57
  @spaces.GPU(duration=3)
58
  def zero_shot_classification(data):
59
- results = classifier(data['sequence'], candidate_labels=data['candidate_labels'], hypothesis_template=data['hypothesis_template'], multi_label=data['multi_label'])
60
- response_string = json.dumps(results)
61
- return response_string
62
 
63
  def create_sequences(data):
64
  # return ['###Given:\n' + data['sequence'] + '\n###End Given\n###Hypothesis:\n' + data['hypothesis_template'].format(label) + "\n###End Hypothesis" for label in data['candidate_labels']]
 
52
  # if 'task' in data and data['task'] == 'few_shot_classification':
53
  # return few_shot_classification(data)
54
  # else:
55
+ return json.dumps(zero_shot_classification(data))
56
 
57
  @spaces.GPU(duration=3)
58
  def zero_shot_classification(data):
59
+ return classifier(data['sequence'], candidate_labels=data['candidate_labels'], hypothesis_template=data['hypothesis_template'], multi_label=data['multi_label'])
 
 
60
 
61
  def create_sequences(data):
62
  # return ['###Given:\n' + data['sequence'] + '\n###End Given\n###Hypothesis:\n' + data['hypothesis_template'].format(label) + "\n###End Hypothesis" for label in data['candidate_labels']]