Ilvir commited on
Commit
22e13e3
·
1 Parent(s): 3c1eab2

Update gpt.py

Browse files
Files changed (1) hide show
  1. gpt.py +1 -1
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([4, 3, 10])
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