Spaces:

IkIzma
/

Transformer_Homework

Runtime error

IkIzma commited on Apr 16, 2023

Commit

fe1596b

1 Parent(s): 7444c03

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ st.markdown("<img width=200px src='https://rozetked.me/images/uploads/dwoilp3BVj
 text = st.text_area("TEXT HERE")
 # ^-- показать текстовое поле. В поле text лежит строка, которая находится там в данный момент
-device = "cuda" if torch.cuda.is_available() else "cpu"
 tok, model = load_tokenizer_and_model("models/essays", device)
 generated = generate(model, tok, "<s>Тема: «Как не застрелиться?». (В.Т. Белинский)\nСочинение: ", device, num_beams=10)

 text = st.text_area("TEXT HERE")
 # ^-- показать текстовое поле. В поле text лежит строка, которая находится там в данный момент
+device = "cpu"
 tok, model = load_tokenizer_and_model("models/essays", device)
 generated = generate(model, tok, "<s>Тема: «Как не застрелиться?». (В.Т. Белинский)\nСочинение: ", device, num_beams=10)