SoniR commited on
Commit
f7ad758
·
1 Parent(s): 613bed0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import streamlit as st
2
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
3
 
4
  # Load custom trained model and tokenizer
5
  model = GPT2LMHeadModel.from_pretrained('https://huggingface.co/spaces/SoniR/chatbotllm/blob/main/config.json')
 
1
  import streamlit as st
2
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
3
+ import https://github.com/codewithresh/pytorch.git
4
 
5
  # Load custom trained model and tokenizer
6
  model = GPT2LMHeadModel.from_pretrained('https://huggingface.co/spaces/SoniR/chatbotllm/blob/main/config.json')