Oshchepkov commited on
Commit
23b856b
1 Parent(s): fb2b8d4

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,7 +70,7 @@ if __name__ == "__main__":
70
  # outputs = model.generate(inputs, max_new_tokens=100, do_sample=False)
71
  # summary = tokenizer.decode(outputs[0], skip_special_tokens=True)
72
  translator = pipeline("summarization", model=m_name,
73
- tokenizer=m_name, max_length=100, device=0
74
  )
75
 
76
  st.subheader('Summary')
 
70
  # outputs = model.generate(inputs, max_new_tokens=100, do_sample=False)
71
  # summary = tokenizer.decode(outputs[0], skip_special_tokens=True)
72
  translator = pipeline("summarization", model=m_name,
73
+ tokenizer=m_name, max_length=100, device=-1
74
  )
75
 
76
  st.subheader('Summary')