paramasivan27 commited on
Commit
b72176a
1 Parent(s): 9051bc2

Update space

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ def ask_question(question, m_tokens):
14
  outputs = model.generate(inputs, attention_mask = attention_mask, max_new_tokens=100, num_return_sequences=1)
15
  gen_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
16
  question, answer = gen_text.split(' A:')
17
- return question, answer
18
  """
19
  For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
20
  """
 
14
  outputs = model.generate(inputs, attention_mask = attention_mask, max_new_tokens=100, num_return_sequences=1)
15
  gen_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
16
  question, answer = gen_text.split(' A:')
17
+ return answer
18
  """
19
  For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
20
  """