Remove phi3 medium.

app.py

@@ -11,8 +11,6 @@ from spacy import displacy
 from spacy.tokens import Span
 import random
 
-json_example = {'nodes': [{'id': 'Aerosmith', 'type': 'organization', 'detailed_type': 'rock band'}, {'id': 'Steven Tyler', 'type': 'person', 'detailed_type': 'lead singer'}, {'id': 'vocal cord injury', 'type': 'medical condition', 'detailed_type': 'fractured larynx'}, {'id': 'retirement', 'type': 'event', 'detailed_type': 'announcement'}, {'id': 'touring', 'type': 'activity', 'detailed_type': 'musical performance'}, {'id': 'September 2023', 'type': 'date', 'detailed_type': 'specific time'}], 'edges': [{'from': 'Aerosmith', 'to': 'Steven Tyler', 'label': 'led by'}, {'from': 'Steven Tyler', 'to': 'vocal cord injury', 'label': 'suffered'}, {'from': 'vocal cord injury', 'to': 'retirement', 'label': 'caused'}, {'from': 'retirement', 'to': 'touring', 'label': 'ended'}, {'from': 'vocal cord injury', 'to': 'September 2023', 'label': 'occurred in'}]}
-
 @spaces.GPU
 def extract(text, model):
     model = Phi3InstructGraph(model=model)    
@@ -74,7 +72,6 @@ def create_custom_entity_viz(data, full_text):
     spans = []
     colors = {}
     for node in data["nodes"]:
-    #   entity_spans = [m.span() for m in re.finditer(re.escape(node["id"]), full_text)]
         entity_spans = find_token_indices(doc, node["id"], full_text)
         for dataentity in entity_spans:
             start = dataentity["start"]
@@ -88,9 +85,6 @@ def create_custom_entity_viz(data, full_text):
                 if node["type"] not in colors:
                     colors[node["type"]] = get_random_light_color()
 
-    # for span in spans:
-    #     print(f"Span: {span.text}, Label: {span.label_}")
-
     doc.set_ents(spans, default="unmodified")
     doc.spans["sc"] = spans
 
@@ -143,9 +137,7 @@ def create_graph(json_data):
         # edge['smooth'] = {'type': 'curvedCW', 'roundness': 0.2}
 
     html = nt.generate_html()
-    # need to remove ' from HTML
     html = html.replace("'", '"')
-    # return html
 
     return f"""<iframe style="width: 140%; height: 620px; margin: 0 auto;" name="result" 
         allow="midi; geolocation; microphone; camera; display-capture; encrypted-media;" 
@@ -158,7 +150,6 @@ def process_and_visualize(text, model):
     if not text or not model:
         raise gr.Error("Text and model must be provided.")
     json_data = extract(text, model)
-    # json_data = json_example
     entities_viz = create_custom_entity_viz(json_data, text)
     
     graph_html = create_graph(json_data)
@@ -173,7 +164,6 @@ with gr.Blocks(title="Phi-3 Instruct Graph (by Emergent Methods") as demo:
         with gr.Column(scale=1):
             input_model = gr.Dropdown(
                 MODEL_LIST, label="Model",
-                # value=MODEL_LIST[0]
             )
             input_text = gr.TextArea(label="Text", info="The text to be extracted")
 

phi3_instruct_graph.py

@@ -13,7 +13,7 @@ login(
 MODEL_LIST = [
   "EmergentMethods/Phi-3-mini-4k-instruct-graph", 
   "EmergentMethods/Phi-3-mini-128k-instruct-graph", 
-  "EmergentMethods/Phi-3-medium-128k-instruct-graph"
+#   "EmergentMethods/Phi-3-medium-128k-instruct-graph"
 ]
 
 torch.random.manual_seed(0)