Update app.py

app.py

@@ -97,12 +97,15 @@ def perform_training():
 
 perform_training()
 
-model.save_pretrained("./fine_tuned_llama2")
-tokenizer.save_pretrained("./fine_tuned_llama2")
+model.save_pretrained("./llama-2-7b-chat_fine_tuned")
+tokenizer.save_pretrained("./llama-2-7b-chat_fine_tuned")
+
+model.push_to_hub("jonACE/llama-2-7b-chat_fine_tuned", token=hf_token)
+tokenizer.push_to_hub("jonACE/llama-2-7b-chat_fine_tuned", token=hf_token)
 
 
 # CHATBOT START
-chatbot = pipeline("text-generation", model="./fine_tuned_llama2")
+chatbot = pipeline("text-generation", model="jonACE/llama-2-7b-chat_fine_tuned")
 
 def chatbot_response(prompt):
     result = chatbot(prompt, max_length=100, do_sample=True, temperature=0.7)