Update main.py

main.py

@@ -104,19 +104,19 @@ async def get_data(input_file: UploadFile = File(...)):
                                                                   
                 Example Output:
                 [
-                    "firstname" : firstname,
-                    "lastname" : lastname,
-                    "contact_number" : contact number
-                    "total_years_of_experience" : total years of experience,
-                    "LinkedIn_link" : LinkedIn link,
-                    "experience" : experience,
+                    "firstname" : "firstname",
+                    "lastname" : "lastname",
+                    "contact_number" : "contact number"
+                    "total_years_of_experience" : "total years of experience",
+                    "LinkedIn_link" : "LinkedIn link",
+                    "experience" : "experience",
                     "skills" : skills
                 ]
                 """
         
         response = model_text.generate_content(prompt)
         print(response.text)
-        data = json.loads(response.text.replace("```json", "").replace("```", ""))
+        data = json.loads(response.text.replace("``` JSON", "").replace("```", ""))
         return {"data": data}
 
     #except Exception as e: