Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-# Load the model and tokenizer
-tokenizer = AutoTokenizer.from_pretrained("potamides/AutomaTikZ")
-model = AutoModelForCausalLM.from_pretrained("potamides/AutomaTikZ")
+# Load the model and tokenizer from the correct repo
+tokenizer = AutoTokenizer.from_pretrained("nllg/tikz-llama-13b")
+model = AutoModelForCausalLM.from_pretrained("nllg/tikz-llama-13b")
 
 def generate_tikz(prompt):
     input_ids = tokenizer(prompt, return_tensors="pt").input_ids