Update main.py

main.py

@@ -35,25 +35,26 @@ def generate_image(prompt, inference_steps, model):
     print(filename)
     
     # Data to be written
-    assertion = {
-                "assertions": [
-                    {
-                    "label": "com.truepic.custom.ai",
-                    "data": {
-                        "model_name": model,
-                        "model_version": "1.0",
-                        "prompt": prompt
-                    }
-                    }
-                ]
-                }
+    # assertion = {
+    #            "assertions": [
+    #                {
+    #                "label": "com.truepic.custom.ai",
+    #                "data": {
+    #                    "model_name": model,
+    #                    "model_version": "1.0",
+    #                    "prompt": prompt
+    #                }
+    #                }
+    #            ]
+    #            }
      
-    json_object = json.dumps(assertion, indent=4)
-    with open("assertion.json", "w") as outfile:
-        outfile.write(json_object)
+    # json_object = json.dumps(assertion, indent=4)
+    # with open("assertion.json", "w") as outfile:
+    #    outfile.write(json_object)
 
     subprocess.check_output(['./truepic-sign', 'init', 'file-system', '--api-key', os.environ.get("api_key")])
-    subprocess.check_output(['./truepic-sign', 'sign', filename, '--assertions', 'assertion.json', '--output', (os.getcwd() + '/static/' + filename)])
+    # subprocess.check_output(['./truepic-sign', 'sign', filename, '--assertions', 'assertion.json', '--output', (os.getcwd() + '/static/' + filename)])
+    subprocess.check_output(['./truepic-sign', 'sign', filename, '--output', (os.getcwd() + '/static/' + filename)])
 
     return {"response": filename}