attilasimko commited on
Commit
0dc48b3
·
1 Parent(s): 85e6b06

single paragraph

Browse files
Files changed (1) hide show
  1. evaluations/models.py +7 -2
evaluations/models.py CHANGED
@@ -2,8 +2,13 @@ from transformers import pipeline
2
  from huggingface_hub import InferenceClient
3
  import os
4
 
5
- system_messages = { "STRICT": "You are a chatbot evaluating github repositories, their python codes and corresponding readme files. Strictly answer the questions with Yes or No.",
6
- "HELP": "You are a chatbot evaluating github repositories, their python codes and corresponding readme files. Please help me answer the following question. Keep your answers short, and informative." }
 
 
 
 
 
7
 
8
  class LocalLLM():
9
  def __init__(self, model_name):
 
2
  from huggingface_hub import InferenceClient
3
  import os
4
 
5
+ system_messages = { "STRICT": """You are a chatbot evaluating github repositories, their python codes and corresponding readme files.
6
+ Strictly answer the questions with "Yes" or "No".
7
+ Don't use any punctuation either.""",
8
+ "HELP": """You are a chatbot evaluating github repositories, their python codes and corresponding readme files.
9
+ Please help me answer the following question.
10
+ Keep your answers short, and informative.
11
+ Your answer should be a single paragraph.""" }
12
 
13
  class LocalLLM():
14
  def __init__(self, model_name):