Update app.py

app.py

@@ -2,13 +2,12 @@ import gradio as gr
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-# Load your model and tokenizer
 from peft import PeftModel, PeftConfig
 from transformers import AutoModelForCausalLM
 
 config = PeftConfig.from_pretrained("AliEssa555/latest-podcast-model-ft")
 base_model = AutoModelForCausalLM.from_pretrained("TheBloke/Mistral-7B-Instruct-v0.2-GPTQ")
-model_name = PeftModel.from_pretrained(base_model, "AliEssa555/latest-podcast-model-ft")
+model = PeftModel.from_pretrained(base_model, "AliEssa555/latest-podcast-model-ft")
 
 #model_name = "path_to_your_fine_tuned_model"  # Use the local path or the Hugging Face model hub ID if published
 model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16)