Spaces:

minnehwg
/

demo-summarization-models

Sleeping

minnehwg commited on May 17

Commit

0a64700

•

1 Parent(s): f9ab7ea

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,6 @@ def processed(text):
     return processed_text
 def summarize(text, model, tokenizer, num_beams=4, device='cpu'):
-    model.eval()
     model.to(device)
     inputs = tokenizer.encode(text, return_tensors="pt", max_length=1024, truncation=True, padding = True).to(device)
     with torch.no_grad():

     return processed_text
 def summarize(text, model, tokenizer, num_beams=4, device='cpu'):
     model.to(device)
     inputs = tokenizer.encode(text, return_tensors="pt", max_length=1024, truncation=True, padding = True).to(device)
     with torch.no_grad():