shashankverma590 commited on
Commit
0210768
·
verified ·
1 Parent(s): cc32d84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ model = AutoModelForSequenceClassification.from_pretrained(model_name)
7
 
8
  classifier = pipeline(task="text-classification", model=model, tokenizer=tokenizer, top_k=None)
9
 
10
- def respond(sentences_json):
11
  sentences = json.loads(sentences_json)
12
  model_outputs = classifier(sentences)
13
  return json.dumps(model_outputs) # produces a list of dicts for each of the labels
 
7
 
8
  classifier = pipeline(task="text-classification", model=model, tokenizer=tokenizer, top_k=None)
9
 
10
+ def get_chatbot_response(sentences_json):
11
  sentences = json.loads(sentences_json)
12
  model_outputs = classifier(sentences)
13
  return json.dumps(model_outputs) # produces a list of dicts for each of the labels