Update main.py

main.py

@@ -24,7 +24,7 @@ def root():
     return {"API": "Sum of 2 Squares"}
     
 @app.post("/img2img")
-async def predict(prompt=Body(...),imgbase64data=Body(...)):
+async def predict(prompt=Body(...),imgbase64data=Body(...),userId=Body(None)):
     pipeline = get_pipeline()
     MAX_QUEUE_SIZE = 4
     start = time.time()
@@ -32,7 +32,7 @@ async def predict(prompt=Body(...),imgbase64data=Body(...)):
     image_data = base64.b64decode(imgbase64data)
     image1 = Image.open(io.BytesIO(image_data))
     w, h = image1.size
-    newW = 256
+    newW = 512
     newH = int(h * newW / w)
     img = image1.resize((newW, newH))  
     end1 = time.time()