Update app.py

app.py

@@ -1,8 +1,8 @@
 # Load model directly
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-tokenizer = AutoTokenizer.from_pretrained("TheCraftySlayer/llama/tree")
-model = AutoModelForCausalLM.from_pretrained("TheCraftySlayer/llama/tree")
+tokenizer = AutoTokenizer.from_pretrained("TheCraftySlayer/llama/tree/main")
+model = AutoModelForCausalLM.from_pretrained("TheCraftySlayer/llama/tree/main")
 input_text =" hello how are you?"
 inputs= tokenizer.encode(input_text, return_tensors='pt')
 outputs= model.generate(inputs,max_length=50,num_return_sequences=5,temperature=0.7)