Upload compare_gradio.py

compare_gradio.py

@@ -21,7 +21,7 @@ API_URL = os.getenv("API_URL")
 API_KEY = os.getenv("API_KEY")
 
 auth_token = os.environ.get("TOKEN") or True
-hf_repo = "Iker/Feedback_FactCheking_cohere"
+hf_repo = "Iker/Feedback_FactCheking_Mar4"
 huggingface_hub.create_repo(
     repo_id=hf_repo,
     repo_type="dataset",
@@ -35,7 +35,7 @@ headers = {"X-API-Key": API_KEY, "Content-Type": "application/json"}
 
 
 def update_models():
-    models = ["pro", "pro-cohere-embeddings", "pro-cohere-reranker"]
+    models = ["pro", "pro-gpt", "pro-rag", "pro-rag-gpt"]
 
     # Randomly select if we want to use pro or turbo models
     # if random.random() < 0.5:
@@ -123,9 +123,7 @@ def format_response(response):
     for q, a in rq.items():
         rq_block.append("**{}**\n{}".format(q, a))
 
-    return "## {}\n\n### {}\n\n{}\n\n{}".format(
-        title, main_claim, fc, "\n".join(rq_block)
-    )
+    return "## {}\n\n### {}\n\n{}\n\n{}".format(title, main_claim, fc, "\n".join(rq_block))
 
 
 def do_both_fact_checking(msg):