MBinAsif commited on
Commit
a256066
·
1 Parent(s): 0274307

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -5,12 +5,12 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer
5
  import streamlit as st
6
 
7
  # Update with your local file paths
8
- model_path = "https://huggingface.co/spaces/MBinAsif/Banking-Chatbot/tree/main"
9
- model = GPT2LMHeadModel.from_pretrained(model_path)
10
 
11
  # Load the fine-tuned GPT-2 model and tokenizer
12
- model = GPT2LMHeadModel.from_pretrained(model)
13
- tokenizer = GPT2Tokenizer.from_pretrained(model)
 
14
 
15
  st.title('Banking Chatbot with GPT-2')
16
 
 
5
  import streamlit as st
6
 
7
  # Update with your local file paths
8
+ model_directory = 'https://huggingface.co/spaces/MBinAsif/Banking-Chatbot/tree/main'
 
9
 
10
  # Load the fine-tuned GPT-2 model and tokenizer
11
+ model = GPT2LMHeadModel.from_pretrained(model_directory)
12
+ tokenizer = GPT2Tokenizer.from_pretrained(model_directory)
13
+
14
 
15
  st.title('Banking Chatbot with GPT-2')
16