update

llm/inference.py

@@ -98,45 +98,45 @@ def extract_info(text):
     print(output)
 
 
-
 def get_name(url, object):
-	messages = [
-		{
-			"role": "user",
-			"content": [
-				{
-					"type": "text",
-					"text": f"Is this a {object}?. Can you guess what it is and give me the closest brand it resembles to? or a model number? And give me its average price in today's market in USD. In output, give me its normal name, model name, model number and price. separated by commas. No description is needed." 
-				},
-				{
-					"type": "image_url",
-					"image_url": {
-						"url": url
-					}
-				}
-			]
-		}
-	]
-
-	completion = client.chat.completions.create(
-		model="meta-llama/Llama-3.2-11B-Vision-Instruct", 
-		messages=messages, 
-		max_tokens=500
-	)
-
-
-	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')
-     
-	result = extract_product_info(llm_result)
-	print(f'\n\nResult brand and price:{result}')
-    
-	# result2 = extract_info(llm_result)
-	# print(f'\n\nFrom Google llm:{result2}')
-
-	return result
+    messages = [
+        {
+            "role": "user",
+            "content": [
+                {
+                    "type": "text",
+                    "text": f"Is this a {object}?. Can you guess what it is and give me the closest brand it resembles to? or a model number? And give me its average price in today's market in USD. In output, give me its normal name, model name, model number and price. separated by commas. No description is needed." 
+                },
+                {
+                    "type": "image_url",
+                    "image_url": {
+                        "url": url
+                    }
+                }
+            ]
+        }
+    ]
+
+    completion = client.chat.completions.create(
+        model="meta-llama/Llama-3.2-11B-Vision-Instruct", 
+        messages=messages, 
+        max_tokens=500
+    )
+
+    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\nThat is the output')
+
+    # 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"