Soumen commited on
Commit
36603f5
·
1 Parent(s): 61ecb4f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ import os
28
  import streamlit as st
29
  import os
30
  import torch
31
- from transformers import AutoTokenizer, AutoModelWithLMHead
32
 
33
 
34
  # NLP Pkgs
 
28
  import streamlit as st
29
  import os
30
  import torch
31
+ from transformers import AutoTokenizer, AutoModelWithLMHead, GPT2LMHeadModel
32
 
33
 
34
  # NLP Pkgs