paramasivan27
commited on
Commit
•
b72176a
1
Parent(s):
9051bc2
Update space
Browse files
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
|
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 |
"""
|