Small updates

app.py

@@ -234,17 +234,17 @@ def process_and_visualize(text, progress=gr.Progress()):
 
 # Example texts in different languages
 EXAMPLES = [
-    [handle_text("""My son Tom, as my direct descendant and John's father, has built a long-standing, companion-like relationship with Brown, significantly influencing each other's growth within the family.
-Mary, my maternal grandmother, has imparted unwavering wisdom and love not only to Brown and me but also to Tom and John, and she has maintained a deep familial bond with Jane's mother.
-Brown, transcending a mere brotherly relationship, has forged a strong father-child bond with Daniel as his biological father, and with Jane, they have significantly impacted each other's lives through numerous family gatherings, forming a solid connection.
-Jane, my cousin, contributes to the family's unity through her informal rapport with Tom, and with Lisa, she shares a special bond that goes beyond ordinary cousin relationships, mutually supporting each other.""")],
+    [handle_text("""The family of Azerbaijan President Ilham Aliyev leads a charmed, glamorous life, thanks in part to financial interests in almost every sector of the economy. His wife, Mehriban, comes from the privileged and powerful Pashayev family that owns banks, insurance and construction companies, a television station and a line of cosmetics. She has led the Heydar Aliyev Foundation, Azerbaijan’s pre-eminent charity behind the construction of schools, hospitals and the country’s major sports complex. Their eldest daughter, Leyla, editor of Baku magazine, and her sister, Arzu, have financial stakes in a firm that won rights to mine for gold in the western village of Chovdar and Azerfon, the country’s largest mobile phone business. Arzu is also a significant shareholder in SW Holding, which controls nearly every operation related to Azerbaijan Airlines (“Azal”), from meals to airport taxis. Both sisters and brother Heydar own property in Dubai valued at roughly $75 million in 2010; Heydar is the legal owner of nine luxury mansions in Dubai purchased for some $44 million.""")],
+    
+    [handle_text("""Legendary rock band Aerosmith has officially announced their retirement from touring after 54 years, citing 
+    lead singer Steven Tyler's unrecoverable vocal cord injury. 
+    The decision comes after months of unsuccessful treatment for Tyler's fractured larynx, 
+    which he suffered in September 2023.""")],
     
     [handle_text("""Pop star Justin Timberlake, 43, had his driver's license suspended by a New York judge during a virtual 
     court hearing on August 2, 2024. The suspension follows Timberlake's arrest for driving while intoxicated (DWI) 
     in Sag Harbor on June 18. Timberlake, who is currently on tour in Europe, 
     pleaded not guilty to the charges.""")],
-
-    [handle_text("""The family of Azerbaijan President Ilham Aliyev leads a charmed, glamorous life, thanks in part to financial interests in almost every sector of the economy. His wife, Mehriban, comes from the privileged and powerful Pashayev family that owns banks, insurance and construction companies, a television station and a line of cosmetics. She has led the Heydar Aliyev Foundation, Azerbaijan’s pre-eminent charity behind the construction of schools, hospitals and the country’s major sports complex. Their eldest daughter, Leyla, editor of Baku magazine, and her sister, Arzu, have financial stakes in a firm that won rights to mine for gold in the western village of Chovdar and Azerfon, the country’s largest mobile phone business. Arzu is also a significant shareholder in SW Holding, which controls nearly every operation related to Azerbaijan Airlines (“Azal”), from meals to airport taxis. Both sisters and brother Heydar own property in Dubai valued at roughly $75 million in 2010; Heydar is the legal owner of nine luxury mansions in Dubai purchased for some $44 million.""")]
 ]
 
 # Function to preprocess the first example when the app starts

phi3_instruct_graph.py

@@ -18,27 +18,8 @@ class Phi3InstructGraph:
     def __init__(self, model = "EmergentMethods/Phi-3-mini-4k-instruct-graph"):
         
         self.model_path = model
-        # self.model = AutoModelForCausalLM.from_pretrained( 
-        #     self.model_path,
-        #     device_map="cuda",  
-        #     torch_dtype="auto",  
-        #     trust_remote_code=True,  
-        # ) 
-        # self.tokenizer = AutoTokenizer.from_pretrained(self.model_path) 
-        # self.pipe = pipeline(
-        #     "text-generation", 
-        #     model=self.model, 
-        #     tokenizer=self.tokenizer, 
-        # )
 
     def _generate(self, messages):
-        # generation_args = { 
-        #     "max_new_tokens": 2000,
-        #     "return_full_text": False,
-        #     "temperature": 0.1,
-        #     "do_sample": False,
-        # } 
-
         # Use the chat_completion method
         response = client.chat_completion(
             messages=messages,
@@ -48,7 +29,6 @@ class Phi3InstructGraph:
         # Access the generated text
         generated_text = response.choices[0].message.content
         return generated_text
-        # return self.pipe(messages, **generation_args)
 
     def _get_messages(self, text):
         context = dedent("""\n