Update app.py

app.py

@@ -16,25 +16,17 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
-def remote_text_encoder(prompts, max_retries=3):
+def remote_text_encoder(prompts):
     from gradio_client import Client
-    import time
     
-    for attempt in range(max_retries):
-        try:
-            client = Client("multimodalart/mistral-text-encoder")
-            result = client.predict(
-                prompt=prompts,
-                api_name="/encode_text"
-            )
-            prompt_embeds = torch.load(result[0])
-            return prompt_embeds
-        except Exception as e:
-            print(f"Text encoder attempt {attempt + 1}/{max_retries} failed: {e}")
-            if attempt < max_retries - 1:
-                time.sleep(2)
-            else:
-                raise Exception(f"Text encoder failed after {max_retries} attempts: {e}")
+    client = Client("multimodalart/mistral-text-encoder")
+    result = client.predict(
+        prompt=prompts,
+        api_name="/encode_text"
+    )
+    
+    prompt_embeds = torch.load(result[0])
+    return prompt_embeds
 
 # Load model
 repo_id = "black-forest-labs/FLUX.2-dev"
@@ -266,3 +258,4 @@ FLUX.2 [dev] is a 32B model rectified flow capable of generating, editing and co
         outputs=[result, seed]
     )
 
+demo.launch(css=css)