Update app.py

app.py

@@ -1,12 +1,14 @@
 import gradio as gr
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
-# LLaMA 3.3 8B Modell und Tokenizer laden
-model_name = "meta-llama/Llama-3.3-8B"
+# Modellname
+model_name = "meta-llama/Llama-3.1-8B-Instruct"
+
+# Tokenizer und Modell laden
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
-    device_map=None,  # Keine GPU-Zuweisung
+    device_map=None,  # Keine GPU
     torch_dtype="float32"  # Float32 für CPU
 )
 
@@ -21,8 +23,8 @@ interface = gr.Interface(
     fn=generate_response,
     inputs="text",
     outputs="text",
-    title="LLaMA 3.3 8B Text Generator (CPU)",
-    description="Gib einen Text ein, und LLaMA 3.3 8B generiert eine Antwort."
+    title="LLaMA 3.1 8B Instruct Text Generator (CPU)",
+    description="Gib einen Text ein, und LLaMA 3.1 8B Instruct generiert eine Antwort."
 )
 
 # App starten