Update app.py

app.py

@@ -10,7 +10,7 @@ import spaces
 
 
 # Model configuration
-MODEL_ID = "oberbics/llama-3.1-newspaper-arguments-grpo-stable-v2"
+MODEL_ID = "oberbics/llama-3.1-argument_mining-v2"
 
 
 SYSTEM_PROMPT = """You are an expert at analyzing historical texts and you hate to summarize
@@ -26,7 +26,6 @@ OUTPUT FORMAT - EXACTLY these 4 XML tags and NOTHING else:
 <explanation>The prefect explicitly argues that existing help is "insufficient" and makes a direct claim that "extraordinary measures" are needed, presenting a clear premise-conclusion structure about the inadequacy of current response.</explanation>
 <human_verification_needed>False</human_verification_needed>
 
-
 EXAMPLE WITHOUT ARGUMENT:
 <argument>NA</argument>
 <claim>NA</claim>
@@ -34,6 +33,9 @@ EXAMPLE WITHOUT ARGUMENT:
 <human_verification_needed>FALSE</human_verification_needed>
 
 RULES:
+- NEVER REPEAT ARGUMENTS, NEVER PRINT ARGUMENTS DOUBLE
+- TRUE FOR VERIFICATION WHEN UNCERTAIN
+- Only output arguments that appear verbatim (or nearly verbatim) in the text
 - NO SUMMARY; ONLY EXACT EXTRACTOM FROM THE TEXT; don't extract anything that is not in the text. Only extract word by word
 - ONLY output these 4 XML tags
 - Extract only original text without changes or use NA when you did not find an argument
@@ -41,12 +43,11 @@ RULES:
 - The CLAIM should say what the (implicite) argument implies, what the main conclusion is
 - Give attention to implicit argumetns
 - In cases of uncertainty or ambiguity, say human_verification_needed TRUE
-- If no argument exists, use NA for all fields except <human_verification_needed>FALSE or TRUE</human_verification_needed>
+- If no argument exists, use NA for ALL fields without explanation except <human_verification_needed>FALSE or TRUE</human_verification_needed>
 - More than one argument possible for one aticle, one unit has one clear clame and all the xml structures
 
 VERIFICATION: BEfore you print the results, double check claims and explanations of the argument. When the claim is just a translation or the explanation states that this is not an argument, don't print it"""
 
-
 print("Loading tokenizer...")
 tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
 tokenizer.pad_token = tokenizer.eos_token