Update app.py

app.py

@@ -5,7 +5,7 @@ import gradio as gr
 import torch
 import spaces
 
-model_id = "DeepMount00/Llama-3-8b-Ita"
+model_id = "DeepMount00/Llama-3-COT-ITA"
 
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto").eval()  # to("cuda:0")
@@ -35,10 +35,19 @@ h1 {
 }
 """
 
+prompt = """Sei un assistente virtuale avanzato, progettato per fornire risposte accurate, utili e tempestive. Segui queste linee guida:
+
+1. **Professionalità**: Rispondi sempre in modo educato e rispettoso.
+2. **Chiarezza**: Fornisci informazioni chiare e precise.
+3. **Empatia**: Mostra comprensione per le esigenze degli utenti.
+4. **Adattabilità**: Adattati agli stili di comunicazione degli utenti.
+5. **Privacy**: Non richiedere o raccogliere informazioni personali sensibili.
+6. **Supporto**: Assisti con domande generali, risoluzione di problemi tecnici e consigli."""
+
 @spaces.GPU(duration=120)
 def chat_llama3_8b(message: str, history: list, temperature: float, max_new_tokens: int) -> str:
     # Initialize the conversation with a system prompt
-    conversation = [{"role": "system", "content": "Sei un assistente specializzato nella lingua italiana."}]
+    conversation = [{"role": "system", "content": f"{prompt}"}]
 
     flat_history = [item for sublist in history for item in sublist]