Update app.py

app.py

@@ -22,14 +22,15 @@ link_dict = {
     "audio classification": "https://huggingface.co/tasks/audio-classification",
     "automatic speech recognition": "https://huggingface.co/tasks/automatic-speech-recognition",
     "fill-mask":"https://huggingface.co/tasks/fill-mask",
+    "image classification": "https://huggingface.co/tasks/image-classification",
+    "image segmentation": "https://huggingface.co/tasks/image-segmentation",
     "question answering":"https://huggingface.co/tasks/question-answering",
     "text-to-speech":"https://huggingface.co/tasks/text-to-speech",
     "sentence similarity": "https://huggingface.co/tasks/sentence-similarity",
     "summarization":"https://huggingface.co/tasks/summarization",
     "text generation": "https://huggingface.co/tasks/text-generation",
     "translation": "https://huggingface.co/tasks/translation",
-    "image classification": "https://huggingface.co/tasks/image-classification",
-    "image segmentation": "https://huggingface.co/tasks/image-segmentation",
+    "token classification": "https://huggingface.co/tasks/token-classification",
     "object detection": "https://huggingface.co/tasks/object-detection"}
     
 
@@ -75,7 +76,7 @@ if input:
      
     with placeholder.container():
         last_message = message_history[-1]
-        if last_message != "":
+        if last_message:
             message(last_message["text"], last_message["is_user"])