fix

main.py

@@ -117,6 +117,7 @@ def generate(request: GenerationRequest):
             os.kill(mainpid, signal.SIGTERM)
         return {"error": str(e)}
 
+'''
 @app.post('/image-to-image')
 def image_to_image(request: GenerationRequest):
     #prompt is a base64 encoded image
@@ -135,7 +136,7 @@ def image_to_image(request: GenerationRequest):
         #get base64 of image
         image = base64.b64encode(image).decode("ascii")
         return GenerationOutput(output=[image])
-        
+'''
 
 @app.post('/generate', response_model=Union[GenerationOutput, ErrorOutput])
 def generate(request: GenerationRequest):