Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
9 |
-
model = GPT2LMHeadModel.from_pretrained(model_path)
|
10 |
|
11 |
# Load the fine-tuned GPT-2 model and tokenizer
|
12 |
-
model = GPT2LMHeadModel.from_pretrained(
|
13 |
-
tokenizer = GPT2Tokenizer.from_pretrained(
|
|
|
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 |
|