Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
-from transformers import GPT2LMHeadModel, GPT2Tokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
 # Initialisierung des Modells und des Tokenizers
-tokenizer = GPT2Tokenizer.from_pretrained("Loewolf/L-GPT_1.1")
-model = GPT2LMHeadModel.from_pretrained("Loewolf/L-GPT_1.1")
+tokenizer = AutoTokenizer.from_pretrained("Atomic-Ai/AtomicGPT_2")
+model = AutoModelForCausalLM.from_pretrained("Atomic-Ai/AtomicGPT_2")
 
 def generate_text(prompt):
     input_ids = tokenizer.encode(prompt, return_tensors="pt")