Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 import os
+import re
 
 # Load API key from environment variables
 HF_API_TOKEN = os.getenv("HUG_TOKEN_READ")
@@ -10,7 +11,10 @@ client = InferenceClient(model="mistralai/Mistral-7B-Instruct-v0.1", token=HF_AP
 
 # Function to translate text into emojis
 def text_to_emoji(text):
-    prompt = f"Convert this sentence into an emoji-sequence of the same meaning and return only the emojis, no explanation:\n\n\"{text}\""
+    # remove special characters
+    text_cleaned = re.sub(r"[.,!?;:]", "", text)
+    
+    prompt = f"Convert this sentence into an emoji-sequence of the same meaning and return only the emojis, no explanation:\n\n\"{text_cleaned}\""
     response = client.text_generation(prompt, max_new_tokens=50)
     return response