Spaces:

thexForce
/

combat_frontend

Running

Junaidb commited on 23 days ago

Commit

bebef8d

verified ·

1 Parent(s): f1aa6aa

Update ui.py

Files changed (1) hide show

ui.py CHANGED Viewed

@@ -15,7 +15,7 @@ def Target_Identification(userinput):
     model = GLiNER.from_pretrained("Ihor/gliner-biomed-bi-small-v1.0")
     labels = ["Protein"]
-    entities = model.predict_entities(text, labels, threshold=0.5)
     result={}

     model = GLiNER.from_pretrained("Ihor/gliner-biomed-bi-small-v1.0")
     labels = ["Protein"]
+    entities = model.predict_entities(userinput, labels, threshold=0.5)
     result={}