Chris4K commited on
Commit
eef7680
1 Parent(s): 197814b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -229,10 +229,11 @@ model_name = 'facebook/blenderbot-400M-distill'
229
  tokenizer = BlenderbotTokenizer.from_pretrained(model_name)
230
  model = BlenderbotForConditionalGeneration.from_pretrained(model_name)
231
 
232
- def func (message, checkbox, numb):
233
  inputs = tokenizer(message, return_tensors="pt")
234
  result = model.generate(**inputs)
235
- return tokenizer.decode(result[0]),"0.9"
 
236
 
237
 
238
 
@@ -273,7 +274,7 @@ def callChains(current_message):
273
  topic_sale_inform_result = topic_sale_inform(current_message)
274
  #conversation.append_response("The Big lebowski.")
275
  #conversation.add_user_input("Is it good?")
276
- final_answer = pipeline_predict_chat(convesation.add_user_input(current_message))
277
  return final_answer, sentiment_analysis_result, topic_sale_inform_result
278
 
279
  chat_bot = gr.Interface(fn=callChains , inputs="textbox", outputs=["textbox","textbox","textbox"], title="Conversation Bot with extra")
 
229
  tokenizer = BlenderbotTokenizer.from_pretrained(model_name)
230
  model = BlenderbotForConditionalGeneration.from_pretrained(model_name)
231
 
232
+ def func (message):
233
  inputs = tokenizer(message, return_tensors="pt")
234
  result = model.generate(**inputs)
235
+ print(result)
236
+ return tokenizer.decode(result[0])
237
 
238
 
239
 
 
274
  topic_sale_inform_result = topic_sale_inform(current_message)
275
  #conversation.append_response("The Big lebowski.")
276
  #conversation.add_user_input("Is it good?")
277
+ final_answer = func(current_message)
278
  return final_answer, sentiment_analysis_result, topic_sale_inform_result
279
 
280
  chat_bot = gr.Interface(fn=callChains , inputs="textbox", outputs=["textbox","textbox","textbox"], title="Conversation Bot with extra")