Update main.py

main.py

@@ -462,8 +462,9 @@ def generate_image(request: Request, input_data: InputImage):
     while attempt < max_attempts:
         try:
             print(input_data.input)
-            if input_data.variante = False: 
-                client = Client("AP123/SDXL-Lightning")
+            if input_data.variante = False:
+                #client = Client("AP123/SDXL-Lightning")
+                client = Client("ByteDance/SDXL-Lightning")
                 result = client.predict(
             		input_data.input,	
             		"8-Step",