Spaces:

Einmalumdiewelt
/

German_text_summarization

Build error

Einmalumdiewelt commited on Aug 10, 2022

Commit

9e77097

1 Parent(s): 268dcad

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ def summarize(inputs,model,summary_length):
         tokenizer = AutoTokenizer.from_pretrained("Einmalumdiewelt/BART_large_CNN_GNAD")
         model = AutoModelForSeq2SeqLM.from_pretrained("Einmalumdiewelt/BART_large_CNN_GNAD")
-  device = "CPU"
   #"cuda" if torch.cuda.is_available() else "CPU"
   model.to(device)

         tokenizer = AutoTokenizer.from_pretrained("Einmalumdiewelt/BART_large_CNN_GNAD")
         model = AutoModelForSeq2SeqLM.from_pretrained("Einmalumdiewelt/BART_large_CNN_GNAD")
+  device = "cpu"
   #"cuda" if torch.cuda.is_available() else "CPU"
   model.to(device)