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