Spaces:
Runtime error
Runtime error
Rzhishchev
commited on
Commit
•
5e40a0a
1
Parent(s):
58d5d8a
Update gpt2.py
Browse files
gpt2.py
CHANGED
@@ -2,6 +2,12 @@ import streamlit as st
|
|
2 |
from transformers import GPT2LMHeadModel, GPT2Tokenizer
|
3 |
import torch
|
4 |
|
|
|
|
|
|
|
|
|
|
|
|
|
5 |
DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
6 |
|
7 |
model_path = "zhvanetsky_model"
|
|
|
2 |
from transformers import GPT2LMHeadModel, GPT2Tokenizer
|
3 |
import torch
|
4 |
|
5 |
+
|
6 |
+
def app():
|
7 |
+
st.title('GPT-2 Text Generator')
|
8 |
+
st.write('This is the GPT-2 text generator page.')
|
9 |
+
|
10 |
+
|
11 |
DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
12 |
|
13 |
model_path = "zhvanetsky_model"
|