Update app.py

app.py

@@ -1,11 +1,14 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
-from huggingface_hub import InferenceClient
+from huggingface_hub import login
+
+# Authenticate to Hugging Face
+login(token="HUGGINGFACE_TOKEN")
 
 # Load the model and tokenizer
-model_name = "your_username/goodnight_storytelling_model"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+model_name = "memn/goodnightde"
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=True)
+model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=True)
 
 def generate_story(prompt):
     inputs = tokenizer(prompt, return_tensors='pt')