andreska commited on
Commit
48097b8
·
verified ·
1 Parent(s): 250f8c1

Removed text return size restrictions

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ context = "Adrega is a very cool company, that implements AI. Rett fra Rio is a
25
  user_input = st.text_input('Ask me a question')
26
  if st.button("Submit"):
27
  if user_input:
28
- answer = pipe(f"Context: {context}\nQuestion: {user_input}\nAnswer:", max_length=512, num_return_sequences=1)
29
  #result = pipe(text_inputs, max_length=200, num_return_sequences=1)[0]['generated_text']
30
  #answer = result.split("Answer:")[1].strip()
31
  #answer = pipe(question=user_input, context=context)
 
25
  user_input = st.text_input('Ask me a question')
26
  if st.button("Submit"):
27
  if user_input:
28
+ answer = pipe(f"Context: {context}\nQuestion: {user_input}\nAnswer:")
29
  #result = pipe(text_inputs, max_length=200, num_return_sequences=1)[0]['generated_text']
30
  #answer = result.split("Answer:")[1].strip()
31
  #answer = pipe(question=user_input, context=context)