Update app.py

app.py

@@ -27,25 +27,26 @@ def load_model():
 
 
 def chatbot_response(prompt):
+    global model, tokenizer  # Assurez-vous que la fonction accède aux variables globales
+    
     inputs = tokenizer(
         alpaca_prompt.format(prompt), 
         return_tensors="pt", 
         truncation=True, 
         max_length=512
     )
-    print(inputs)
+    
     outputs = model.generate(
         inputs["input_ids"],
         max_new_tokens=1024,
         temperature=0.7,
         do_sample=True
     )
-    print(outputs)
     
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    print(response)
     reponse = response.split("### Response:")
-    return reponse[-1]
+    
+    return reponse[-1].strip()  # Supprime les espaces vides inutiles
 
 # Load model and tokenizer
 model, tokenizer = load_model()