Update app.py

app.py

@@ -5,7 +5,8 @@ from ollama import chat
 from ollama import ChatResponse
 
 # Default model
-OLLAMA_MODEL = "llama3.2:3b"
+OLLAMA_MODEL = "llama3.2:3b-instruct-q3_K_M"
+# OLLAMA_MODEL = "llama3.2:1b"
 
 # Load BERT MODEL
 from transformers import pipeline, DistilBertTokenizerFast
@@ -57,15 +58,14 @@ AITA : Am I the asshole? Usually posted in the question.
 YTA : You are the asshole in this situation.  
 NTA : You are not the asshole in this situation.
 
-### The task for you predict if most of the users would tag the given situation as YTA or NTA, give your personal opinion. Do not try to be nice, just give brutally honest and unbiased view. Base your decision entirely on the given text. 
-Use second person terms like you in the explanation.
- 
+### The task for you label YTA or NTA for the given text. Give a short explanation for the label. Be brutally honest and unbiased. Base your explanation entirely on the given text only. 
+
+If the label is YTA, also explain what could the user have done better.  
 ### The output format is as follows:
 "YTA" or "NTA", a short explanation. 
 
-
 ### Situation :  {question}
-### Output :{expected_class}"""
+### Response :"""
 
     explain_only_prompt =  f"""
 ### You know about the subreddit community r/AmItheAsshole. In this community people post their life situations and ask if they are the asshole or not. 
@@ -74,12 +74,9 @@ AITA : Am I the asshole? Usually posted in the question.
 YTA : You are the asshole in this situation.  
 NTA : You are not the asshole in this situation.
 
-### The task for you explain why a particular situation was tagged as NTA or YTA by most users. I will give the situation as well as the NTA or YTA tag. just give your explanation for the label.  Do not try to be nice, just give brutally honest and unbiased view. Base your decision entirely on the given text and the label tag. 
+### The task for you explain why a particular situation was tagged as NTA or YTA by most users. I will give the situation as well as the NTA or YTA tag. just give your explanation for the label. Be nice but give a brutally honest and unbiased view. Base your explanation entirely on the given text and the label tag only. Do not assume anything extra.  
 Use second person terms like you in the explanation.
 
-### The output format is just the explanation for the label. 
-
-
 ### Situation :  {question}
 ### Label Tag : {expected_class}
 ### Explanation for {expected_class} :"""
@@ -101,7 +98,7 @@ Use second person terms like you in the explanation.
 
 def gradio_interface(prompt, selected_model):
     if selected_model == MODEL_CHOICE_LLAMA:
-        response = ask_ollama(prompt, selected_model)
+        response = ask_ollama(prompt)
     elif selected_model == MODEL_CHOICE_BERT:
         response, confidence = ask_bert(prompt)
         response = f"{response} with confidence {confidence}"