Update app.py

app.py

@@ -2,8 +2,8 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 # טעינת המודל וה-tokenizer
-model = AutoModelForCausalLM.from_pretrained("gpt2")
-tokenizer = AutoTokenizer.from_pretrained("gpt2")
+model = AutoModelForCausalLM.from_pretrained("distilgpt2")
+tokenizer = AutoTokenizer.from_pretrained("distilgpt2")
 
 # פונקציה שמקבלת פרומפט ומחזירה תגובה
 def generate_text(prompt):