Update ui.py

ui.py

@@ -1,6 +1,7 @@
 import streamlit as st 
 import requests
-import os 
+import os
+from gliner import GLiNER
 #from auth import authenticator
 
 tok=os.getenv("TOK")
@@ -8,10 +9,20 @@ tok=os.getenv("TOK")
 
 
 
+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={}
+    for entity in entities:
+        if entity["label"] =="Protein":
+            return entity["text"]
 
-               
 
 def APP():
             
@@ -58,8 +69,9 @@ def APP():
                         with st.form("bio",border=False):
 
                             project_name=None
-                            target=None
+                           
 
+                            
 
                             def scan_for_project_availability(user_id):
                                 request_url=f"https://thexforce-combat-backend.hf.space/{user_id}/projects"
@@ -103,7 +115,7 @@ def APP():
                                         payload={
                                             "uid":st.user.email ,
                                             "pid":project_name or projectname,
-                                            "target":target,
+                                            "target":Target_Identification(bio_input) or None,
                                             "high_level_bio_query":bio_input
                                         }