Fix model loading to use Hugging Face Hub repo ID

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 # Load the fine-tuned model and tokenizer
-model_name = "./fine-tuned-distilgpt2"  # Path to the saved fine-tuned model
+model_name = "johnnymullaney/fine-tuned-distilgpt2-books"  # Path to the saved fine-tuned model
 
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)