Update gpt.py
Browse files
gpt.py
CHANGED
@@ -13,7 +13,7 @@ model = GPT2LMHeadModel.from_pretrained(
|
|
13 |
model.load_state_dict(torch.load('modelgpt.pt', map_location=torch.device('cpu')))
|
14 |
|
15 |
|
16 |
-
col1, col2, col3 = st.columns([
|
17 |
|
18 |
with col1:
|
19 |
|
|
|
13 |
model.load_state_dict(torch.load('modelgpt.pt', map_location=torch.device('cpu')))
|
14 |
|
15 |
|
16 |
+
col1, col2, col3 = st.columns([9, 1, 9])
|
17 |
|
18 |
with col1:
|
19 |
|