Update app.py

app.py

@@ -1,6 +1,8 @@
 from datasets import Dataset
 from transformers import AutoTokenizer, AutoModelForCausalLM, TrainingArguments, Trainer
 from datafile import QL, resume_data_dict
+import streamlit as st
+
 
 # Load your custom data
 
@@ -55,6 +57,11 @@ trainer = Trainer(
 # Train the model
 trainer.train()
 
+model_name = "./resume_bot"  # Path to your fine-tuned model
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+
+
 st.title("Resume Chatbot")
 
 if 'history' not in st.session_state: