EmreYY20 commited on
Commit
989abb3
1 Parent(s): ccd2173
Files changed (2) hide show
  1. abstractive_model.py +1 -0
  2. requirements.txt +0 -0
abstractive_model.py CHANGED
@@ -1,4 +1,5 @@
1
  from transformers import pipeline
 
2
 
3
  def summarize_with_bart(text):
4
  pipe = pipeline("text2text-generation", model="EE21/Fine-tuned-BART-ToS-Summarization")
 
1
  from transformers import pipeline
2
+ import torch
3
 
4
  def summarize_with_bart(text):
5
  pipe = pipeline("text2text-generation", model="EE21/Fine-tuned-BART-ToS-Summarization")
requirements.txt CHANGED
Binary files a/requirements.txt and b/requirements.txt differ