IkIzma commited on
Commit
7444c03
·
1 Parent(s): 593bf11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,5 +11,6 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
11
  tok, model = load_tokenizer_and_model("models/essays", device)
12
  generated = generate(model, tok, "<s>Тема: «Как не застрелиться?». (В.Т. Белинский)\nСочинение: ", device, num_beams=10)
13
 
14
- st.markdown(f"{generated[0]}!")
 
15
  # выводим результаты модели в текстовое поле, на потеху пользователю
 
11
  tok, model = load_tokenizer_and_model("models/essays", device)
12
  generated = generate(model, tok, "<s>Тема: «Как не застрелиться?». (В.Т. Белинский)\nСочинение: ", device, num_beams=10)
13
 
14
+ st.markdown(f"Hello!\n")
15
+ st.markdown(generated[0])
16
  # выводим результаты модели в текстовое поле, на потеху пользователю