Update main.py

main.py

@@ -116,7 +116,7 @@ async def get_data(input_file: UploadFile = File(...)):
         
         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("json", "").replace("```", ""))
         return {"data": data}
 
     #except Exception as e: