Update src/v2_for_hf.py

src/v2_for_hf.py

@@ -75,12 +75,4 @@ def generate_images(oai_key, input_path, mistaken_class, ground_truth_class):
         # refined_image.save(output_path + "{}.png".format(i), 'PNG')
         out_images.append(refined_image)
 
-    return tuple(out_images)
-
-
-if __name__ == "__main__":
-    oai_key = "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"
-    input_path = "out/0.png"
-    mistaken_class = "dog"
-    ground_truth_class = "cat"
-    generate_images(oai_key, input_path, mistaken_class, ground_truth_class)
+    return tuple(out_images)