Fix medium phi3

app.py

@@ -80,7 +80,6 @@ def create_custom_entity_viz(data, full_text):
             start = dataentity["start"]
             end = dataentity["end"]
             
-            print("entity spans:", entity_spans)
             if start < len(doc) and end <= len(doc):
                 span = Span(doc, start, end, label=node["type"])
                 
@@ -89,8 +88,8 @@ 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_}")
+    # for span in spans:
+    #     print(f"Span: {span.text}, Label: {span.label_}")
 
     doc.set_ents(spans, default="unmodified")
     doc.spans["sc"] = spans
@@ -160,7 +159,6 @@ def process_and_visualize(text, model):
         raise gr.Error("Text and model must be provided.")
     json_data = extract(text, model)
     # json_data = json_example
-    print(json_data)
     entities_viz = create_custom_entity_viz(json_data, text)
     
     graph_html = create_graph(json_data)

phi3_instruct_graph.py

@@ -83,7 +83,7 @@ class Phi3InstructGraph:
                     -------Text end-------
                     """)
         
-        if self.model == "EmergentMethods/Phi-3-medium-128k-instruct-graph":
+        if self.model_path == "EmergentMethods/Phi-3-medium-128k-instruct-graph":
             # model without system message
             messages = [ 
                 {
@@ -109,6 +109,5 @@ class Phi3InstructGraph:
     def extract(self, text):
         messages = self._get_messages(text)
         pipe_output = self._generate(messages)
-        print("pipe_output", pipe_output)
         print("pipe_output json", pipe_output[0]["generated_text"])
         return pipe_output[0]["generated_text"]