Update app.py

app.py

@@ -24,10 +24,8 @@ logger = logging.getLogger("mistral-text-encoding-gradio")
 # ------------------------------------------------------
 # Config
 # ------------------------------------------------------
-TEXT_ENCODER_ID = os.getenv("TEXT_ENCODER_ID", "mistralai/Mistral-Small-3.2-24B-Instruct-2506")
-TOKENIZER_ID = os.getenv(
-    "TOKENIZER_ID", "mistralai/Mistral-Small-3.1-24B-Instruct-2503"
-)
+TEXT_ENCODER_ID="Qwen/Qwen2.5-7B-Instruct-1M"
+TOKENIZER_ID = "Qwen/Qwen2.5-7B-Instruct-1M"
 DTYPE = torch.bfloat16
 
 # ------------------------------------------------------
@@ -40,11 +38,6 @@ text_encoder = Mistral3ForConditionalGeneration.from_pretrained(
     TEXT_ENCODER_ID,
     dtype=DTYPE,
 ).to("cpu")
-logger.info(
-    "Loaded Mistral text encoder (%.2fs) dtype=%s device=%s",
-    time.time() - t0,
-    text_encoder.dtype,
-)
 
 t1 = time.time()
 tokenizer = AutoProcessor.from_pretrained(TOKENIZER_ID)
@@ -90,7 +83,6 @@ def encode_text(prompt: str):
     )
 
     duration = (time.time() - t0) * 1000.0
-
     logger.info(
         "Encoded in %.2f ms | prompt_embeds.shape=%s | text_ids.shape=%s",
         duration,
@@ -116,7 +108,7 @@ def encode_text(prompt: str):
     )
 
     return temp_file.name, status
-
+    
 
 # ------------------------------------------------------
 # Gradio Interface