Update app.py

app.py

@@ -55,14 +55,10 @@ def infer(prompt):
      #       else:
      #           images.append(image)
     #else:
-    url = os.getenv('JAX_BACKEND_URL')
-    payload = {'prompt': prompt}
-    images_request = requests.post(url, json = payload)
-    for image in images_request.json()["images"]:
-        image_b64 = (f"data:image/jpeg;base64,{image}")
-        images.append(image_b64)
+    with autocast("cuda"):
+        generated = pipe(prompt, guidance_scale=7.5)["sample"]
     
-    return images
+    return generated
     
     
 css = """