sss

llm/inference.py

@@ -69,6 +69,7 @@ def extract_info(text):
 
 
 def get_name(url, object):
+    
 	messages = [
 		{
 			"role": "user",
@@ -94,19 +95,18 @@ def get_name(url, object):
 	)
 
 
-	print(f'\n\nNow output of LLM:\n')
-	llm_result = completion.choices[0].message['content']
-	print(llm_result)
-	# print(f'\n\nThat is the output')
-
+    print(f'\n\nNow output of LLM:\n')
+    llm_result = completion.choices[0].message['content']
+    print(llm_result)
+    # print(f'\n\nThat is the output')
     print(f"Extracting from the output now, function calling")
     result = extract_product_info(llm_result)
-	print(f'\n\nResult brand and price:{result}')
+    print(f'\n\nResult brand and price:{result}')
     print(f'\n\nThat is the output')
-	# result2 = extract_info(llm_result)
-	# print(f'\n\nFrom Google llm:{result2}')
-
-	return result
+    # result2 = extract_info(llm_result)
+    # print(f'\n\nFrom Google llm:{result2}')
+    
+    return result
 
 # url = "https://i.ibb.co/mNYvqDL/crop_39.jpg"
 # object="fridge"