Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import random
+import os
 from llama_cpp import Llama
 
 # Liste des modèles disponibles
@@ -16,15 +17,16 @@ model_versions = {
 llm = None
 current_model = "WizzGPTv7"
 
-def load_model(model_name):
-    global llm, current_model
-    current_model = model_name
-    llm = Llama(
-        model_path=model_versions[model_name],
-        n_threads=2,
-        n_ctx=1024,
-        use_mlock=True
-    )
+model_path = os.path.join("model", model_name)
+if not os.path.exists(model_path):
+    raise FileNotFoundError(f"Model file not found at {model_path}")
+llm = Llama(
+    model_path=model_path,
+    n_ctx=2048,
+    n_threads=2,
+    n_batch=32,
+    verbose=False
+)
 
 # Charger modèle initial
 load_model(current_model)