Update app.py

app.py

@@ -18,7 +18,7 @@ from transformers import pipeline
 
 MAX_CONTEXT_WINDOW_TOKENS = 32000
 MODEL = "speakleash/bielik-11b-v2.6-instruct"
-GUARD_MODEL = "bielik-guard"
+GUARD_MODEL = "speakleash/Bielik-Guard-0.1B-v1.0"
 EMBEDDING_MODEL = "baai/bge-m3"
 TOP_K = 5
 GUARD_THRESHOLD = 0.5
@@ -156,7 +156,7 @@ async def ask_files(message, files):
 
 @cl.step(name="klasyfikowanie", type="tool", show_input=False)
 async def classify(message):
-    classifier = pipeline("text-classification", model=GUARD_MODEL, tokenizer=GUARD_MODEL, local_files_only=True, return_all_scores=True)
+    classifier = pipeline("text-classification", model=GUARD_MODEL, return_all_scores=True)
     results = classifier(message)[0]
     return max(result["score"] for result in results)