Hugging Face's logo

Spaces:
Einmalumdiewelt
/
German_text_summarization
Build error

App Files Community
Einmalumdiewelt commited on
Commit
37cdd8d
·
1 Parent(s): 509e887

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,9 +1,11 @@
1
  import gradio as gr
 
2
 
3
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
4
 
5
  tokenizer = AutoTokenizer.from_pretrained("Einmalumdiewelt/T5-Base_GNAD")
6
  model = AutoModelForSeq2SeqLM.from_pretrained("Einmalumdiewelt/T5-Base_GNAD")
 
7
  model.to(device)
8
 
9
 
 
1
  import gradio as gr
2
+ import torch
3
 
4
  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