Update app.py

app.py

@@ -10,7 +10,12 @@ def text_to_emoji(text):
     # Generate emoji output
     result = generator(prompt, max_length=50, num_return_sequences=1)
     
-    emoji_output = result[0]['generated_text'].split("->")[-1].strip()
+    # Extract the emoji part from the generated text
+    emoji_output = result[0]['generated_text']
+    
+    # Clean the output, removing unnecessary parts and keeping only emojis
+    emoji_output = emoji_output.split("->")[-1].strip()
+    
     return emoji_output
 
 # Gradio Interface
@@ -20,4 +25,4 @@ iface = gr.Interface(fn=text_to_emoji,
                      title="Text-to-Emoji AI",
                      description="Enter a sentence and get the emoji equivalent!")
 
-iface.launch()
+iface.launch()