change to nebius inference

app.py

@@ -19,7 +19,11 @@ HF_API_TOKEN = os.getenv("HUG_TOKEN_READ2")
 #client = InferenceClient(model="openGPT-X/Teuken-7B-instruct-commercial-v0.4", token=HF_API_TOKEN)
 #client = InferenceClient(model=qwenModel, token=HF_API_TOKEN)
 #client = InferenceClient(model="microsoft/Phi-4-mini-instruct", token=HF_API_TOKEN)
-client = InferenceClient(model="openai-community/gpt2", token=HF_API_TOKEN)
+#client = InferenceClient(model="openai-community/gpt2", token=HF_API_TOKEN)
+client = InferenceClient(
+    provider="nebius",
+    api_key="hf_xxxxxxxxxxxxxxxxxxxxxxxx",
+)
 
 # Function to translate text into emojis
 def text_to_emoji(text):
@@ -27,8 +31,26 @@ def text_to_emoji(text):
     text_cleaned = re.sub(r"[.,!?;:]", "", text)
     
     prompt = f"Convert this sentence into an emoji-sequence which conveys a similar meaning and return only the emojis, no explanation:\n\n\"{text_cleaned}\""
-    response = client.text_generation(prompt, max_new_tokens=50)
-    return response
+    #response = client.text_generation(prompt, max_new_tokens=50)
+    #return response
+
+
+completion = client.chat.completions.create(
+    model="mistralai/Mistral-Small-3.1-24B-Instruct-2503",
+    messages=[
+        {
+            "role": "user",
+            "content": [
+                {
+                    "type": "text",
+                    "text": prompt
+                }
+            ]
+        }
+    ],
+    max_tokens=25,
+)
+    return completion.choices[0].message
 
 # Gradio UI
 iface = gr.Interface(