Hugging Face's logo

Spaces:
BigSalmon
/
GPT2_Most_Probable
Runtime error

App Files Community
BigSalmon commited on
Commit
e1311b9
·
1 Parent(s): d633e6a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import torch
6
  import torch.nn as nn
7
  from transformers import ElectraModel, AutoConfig, GPT2LMHeadModel
8
  from transformers.activations import get_activation
9
- from transformers import AutoTokenizer
10
 
11
 
12
  st.title('KoGPT2 Demo')
 
6
  import torch.nn as nn
7
  from transformers import ElectraModel, AutoConfig, GPT2LMHeadModel
8
  from transformers.activations import get_activation
9
+ from transformers import AutoTokenizer, AutoModelWithLMHead
10
 
11
 
12
  st.title('KoGPT2 Demo')