Update utility/utils.py

utility/utils.py

@@ -186,8 +186,7 @@ def extract_text_from_images(image_paths):
 # Function to call the Gemma model and process the output as Json 
 def Data_Extractor(data, client=client):
     text = f'''<s>[INST] Act as a Text extractor for the following text given in text: {data}. 
-    Your task is to extract specific information and return it in a JSON format as outlined below:
-    
+    Your task is to extract specific information and return it in a JSON format as outlined below:    
     Please extract the following details:
     {{
         "Name": ["Identify and extract all the person's names from the text."],
@@ -197,12 +196,12 @@ def Data_Extractor(data, client=client):
         "Address": ["Extract all full postal addresses or locations mentioned in the text."],
         "Email": ["Identify and extract all valid email addresses mentioned in the text; if none are found, return 'Not found'."],
         "Link": ["Identify and extract any website URLs or social media links present in the text."]
-    }}
-    
+    }}    
     Output: 
     [/INST]</s> [INST] 
     Return the extracted information in JSON format as follows: 
-    [/INST]'''
+    [/INST]
+    '''
     
     # Call the API for inference
     response = client.text_generation(text, max_new_tokens=1000)#, temperature=0.4, top_k=50, top_p=0.9, repetition_penalty=1.2)