Update app.py

app.py

@@ -21,19 +21,16 @@ MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
 def remote_text_encoder(prompts):
-    response = requests.post(
-        "https://remote-text-encoder-flux-2.huggingface.co/predict",
-        json={"prompt": prompts},
-        headers={
-            "Authorization": f"Bearer {os.environ['HF_TOKEN']}",
-            "Content-Type": "application/json"
-        }
+    from gradio_client import Client
+    
+    client = Client("multimodalart/mistral-text-encoder")
+    result = client.predict(
+        prompt=prompts,
+        api_name="/encode_text"
     )
-
-    assert response.status_code == 200, f"{response.status_code=}"
-
-    prompt_embeds = torch.load(io.BytesIO(response.content))
-
+    
+    prompt_embeds = torch.load(result[0])
+    
     return prompt_embeds
 
 # Load model