Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
0dc48b3
1
Parent(s):
85e6b06
single paragraph
Browse files- 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.
|
6 |
-
|
|
|
|
|
|
|
|
|
|
|
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):
|