Saiteja Solleti commited on
Commit
8048d7d
·
1 Parent(s): 0ed4471

correcting scores

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -42,6 +42,13 @@ def EvaluateRAGModel(question, evaluation_model):
42
 
43
  query = question.strip()
44
 
 
 
 
 
 
 
 
45
  #invoke create milvus db function
46
  try:
47
  db_collection = CreateMilvusDbSchema()
 
42
 
43
  query = question.strip()
44
 
45
+ if evaluation_model == "LLaMA 3.3":
46
+ EVAL_MODEL = "llama-3.3-70b-specdec"
47
+ PROMPT_MODEL = "llama-3.3-70b-specdec"
48
+ elif evaluation_model == "Mistral 7B":
49
+ EVAL_MODEL = "mixtral-8x7b-32768"
50
+ PROMPT_MODEL = "mixtral-8x7b-32768"
51
+
52
  #invoke create milvus db function
53
  try:
54
  db_collection = CreateMilvusDbSchema()