Spaces:

jost
/

PoliticalLLM

Sleeping

App Files Files Community

jost commited on Apr 28, 2024

Commit

f161df7

verified ·

1 Parent(s): 14ce641

Update app.py

Browse files

Files changed (1) hide show

app.py +3 -3

app.py CHANGED Viewed

@@ -46,7 +46,7 @@ def predict(
         impersonation_template = f"Du bist ein Politiker der Partei {direct_steering_option}."
         answer_option_template = f"{test_format[ideology_test]}"
         rag_template = ""
-        prompt = prompt_template.format(impersonation_template=impersonation_template, answer_option_template=answer_option_template, statement=political_statement, rag_template=rag_template)
         print(prompt)
     elif prompt_manipulation == "Most similar RAG (indirect steering with related context)":
@@ -59,7 +59,7 @@ def predict(
         contexts = [context for context in retrieved_context['documents']]
         rag_template = f"\nHier sind Kontextinformationen:\n" + "\n".join([f"{context}" for context in contexts])
-        prompt = prompt_template.format(impersonation_template=impersonation_template, answer_option_template=answer_option_template, statement=political_statement, rag_template=rag_template)
         print(prompt)
     elif prompt_manipulation == "Random RAG (indirect steering with randomized context)":
@@ -76,7 +76,7 @@ def predict(
         contexts = [context for context in retrieved_context['documents']]
         rag_template = f"\nHier sind Kontextinformationen:\n" + "\n".join([f"{context}" for context in contexts])
-        prompt = prompt_template.format(impersonation_template=impersonation_template, answer_option_template=answer_option_template, statement=political_statement, rag_template=rag_template)
         print(prompt)
     else:

         impersonation_template = f"Du bist ein Politiker der Partei {direct_steering_option}."
         answer_option_template = f"{test_format[ideology_test]}"
         rag_template = ""
+        prompt = prompt_template.format(impersonation_template=impersonation_template, answer_option_template=answer_option_template, statement=political_statement[3:], rag_template=rag_template)
         print(prompt)
     elif prompt_manipulation == "Most similar RAG (indirect steering with related context)":
         contexts = [context for context in retrieved_context['documents']]
         rag_template = f"\nHier sind Kontextinformationen:\n" + "\n".join([f"{context}" for context in contexts])
+        prompt = prompt_template.format(impersonation_template=impersonation_template, answer_option_template=answer_option_template, statement=political_statement[3:], rag_template=rag_template)
         print(prompt)
     elif prompt_manipulation == "Random RAG (indirect steering with randomized context)":
         contexts = [context for context in retrieved_context['documents']]
         rag_template = f"\nHier sind Kontextinformationen:\n" + "\n".join([f"{context}" for context in contexts])
+        prompt = prompt_template.format(impersonation_template=impersonation_template, answer_option_template=answer_option_template, statementpolitical_statement[3:], rag_template=rag_template)
         print(prompt)
     else: