little fixes

Rock_fact_checker.py

@@ -60,7 +60,7 @@ def main():
         # Unfortunately necessary as the Random statement button is _below_ the textbox
         # Adapted for Streamlit>=1.12
         if hasattr(st, 'scriptrunner'):
-            raise st.scriptrunner.script_runner.RerunException(st.scriptrunner.script_requests.RerunData(None))
+            raise st.scriptrunner.script_runner.RerunException(st.scriptrunner.script_requests.RerunData(""))
         else:
             raise st.runtime.scriptrunner.script_runner.RerunException(
             st.runtime.scriptrunner.script_requests.RerunData("") 

app_utils/backend_utils.py

@@ -84,8 +84,8 @@ def query(statement: str, retriever_top_k: int = 5):
             break
 
     results["agg_entailment_info"] = {
-        "contradiction": round(agg_con / scores, 2),
-        "neutral": round(agg_neu / scores, 2),
-        "entailment": round(agg_ent / scores, 2),
+        "contradiction": float(round(agg_con / scores, 2)),
+        "neutral": float(round(agg_neu / scores, 2)),
+        "entailment": float(round(agg_ent / scores, 2)),
     }
     return results

app_utils/config.py

@@ -9,8 +9,9 @@ RETRIEVER_MODEL = "sentence-transformers/msmarco-distilbert-base-tas-b"
 RETRIEVER_MODEL_FORMAT = "sentence_transformers"
 RETRIEVER_TOP_K = 5
 
-try:
+# In HF Space, we use microsoft/deberta-v2-xlarge-mnli
+# for local testing, a smaller model is better
+NLI_MODEL = "valhalla/distilbart-mnli-12-1"
+if hasattr(st, 'secrets'):
     NLI_MODEL = st.secrets['NLI_MODEL']
-except:
-    NLI_MODEL = "valhalla/distilbart-mnli-12-1"
 print(f'Used NLI model: {NLI_MODEL}')

data/statements.txt

@@ -38,7 +38,7 @@ Dave Grohl played in Nirvana and Foo Fighters
 John Michael Stipe was part of the R.E.M.
 Eric Clapton did some gigs with Dire Straits
 Mark Knopfler plays drums
-"La Bamba" is a hit by Ritchie Valens
+Ritchie Valens sang "La Bamba"
 Nico took part in the Velvet Underground
 David Bowie appears in Twin Peaks
 The Cure made dark songs