Update app/main.py

app/main.py

@@ -18,7 +18,7 @@ app.add_middleware(
 )
 
 # Connect to the model
-client = Client("K00B404/flux_666")
+client = Client("NihalGazi/FLUX-Pro-Unlimited")
 
 @app.get("/")
 def root():
@@ -40,18 +40,13 @@ def generate_image(
 ):
     try:
         result = client.predict(
-            prompt=prompt,
-            basemodel=basemodel,
-            width=width,
-            height=height,
-            scales=scales,
-            steps=steps,
-            seed=seed,
-            upscale_factor=upscale_factor,
-            process_upscale=process_upscale,
-            lora_model=lora_model,
-            process_lora=process_lora,
-            api_name="/gen"
+    		prompt=prompt,
+    		width=1280,
+    		height=1280,
+    		seed=0,
+    		randomize=True,
+    		server_choice="NSFW-Core: Uncensored Server 2",
+    		api_name="/generate_image"
         )
 
         # result is a list of local file paths