Update app.py

app.py

@@ -3,8 +3,8 @@ import gradio as gr
 import torch
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-tokenizer = AutoTokenizer.from_pretrained("Norod78/lora-norod78_hebrew-gpt_neo-xl-qa", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("Norod78/lora-norod78_hebrew-gpt_neo-xl-qa", pad_token_id=tokenizer.eos_token_id)
+tokenizer = AutoTokenizer.from_pretrained("Norod78/hebrew-gpt_neo-xl", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("Norod78/hebrew-gpt_neo-xl", pad_token_id=tokenizer.eos_token_id)
 model.to(device)
 
 def chat(message, history=[]):