Hugging Face's logo

Spaces:
Einmalumdiewelt
/
German_text_summarization
Build error

App Files Community
Einmalumdiewelt commited on
Commit
2d14595
·
1 Parent(s): 8287113

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -5,7 +5,8 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
5
 
6
  tokenizer = AutoTokenizer.from_pretrained("Einmalumdiewelt/T5-Base_GNAD")
7
  model = AutoModelForSeq2SeqLM.from_pretrained("Einmalumdiewelt/T5-Base_GNAD")
8
- device = "cuda" if torch.cuda.is_available() else "cpu"
 
9
  model.to(device)
10
 
11
 
 
5
 
6
  tokenizer = AutoTokenizer.from_pretrained("Einmalumdiewelt/T5-Base_GNAD")
7
  model = AutoModelForSeq2SeqLM.from_pretrained("Einmalumdiewelt/T5-Base_GNAD")
8
+ device = "cpu"
9
+ #"cuda" if torch.cuda.is_available() else "cpu"
10
  model.to(device)
11
 
12