Update models.py

models.py

@@ -62,6 +62,38 @@ Text: "{sentence}"
 Please revise this text such that it maintains the criticism in the original text and delivers it in a friendly but professional manner. Make minimal changes to the original text. [/INST] Revised Text: """
     return prompt
 
+def phi_score(sentence):
+    prompt = f"""<|system|>
+You are an assistant helping with paper reviews. 
+You will be tasked to classify sentences as 'J' or 'V'
+
+'J' is positive or 'J' is encouraging. 
+'J' has a neutral tone or 'J' is not discouraging and not encouraging.
+'V' is overly blunt or 'V' contains excessive negativity and no constructive feedback.
+'V' contains an accusatory tone or 'V' contains sweeping generalizations or 'V' contains personal attacks.<|end|> 
+<|user|>
+Text: "{sentence}"
+
+Please classify this text as either 'J' or 'V'. Only output 'J' or 'V' with no additional explanation.<|end|>
+<|assistant|>
+"""
+    return prompt
+
+def phi_revise(sentence):
+    prompt = f"""<|system|>
+You are an assistant that helps users revise Paper Reviews.
+Paper reviews exist to provide authors of academic research papers constructive critism.<|end|>
+<|user|>
+This is text found in a review.
+This text was classified as 'toxic'
+
+Text: "{sentence}"
+
+Please revise this text such that it maintains the criticism in the original text and delivers it in a friendly but professional manner. Make minimal changes to the original text.<|end|>
+<|assistant|>
+"""
+    return prompt
+
 def query_model_score(sentence, api_key, model_id, prompt_fun):
     API_URL = f"https://api-inference.huggingface.co/models/{model_id}"
     headers = {"Authorization": f"Bearer {api_key}"}
@@ -117,6 +149,9 @@ def revise_review(review, api_key, model_id, highlight_color):
     elif 'mistral' in model_id.lower():
         revision_prompt = mistral_revise
         score_prompt = mistral_score
+    elif 'phi' in model_id.lower():
+        revision_prompt = phi_revise
+        score_prompt = phi_score
 
     try:
         review = review.replace('"', "'")