Update app.py

app.py

@@ -1,14 +1,11 @@
-import time
-from threading import Thread
-
 import gradio as gr
 import torch
+
+from threading import Thread
 from PIL import Image
-#from transformers import AutoProcessor, LlavaForConditionalGeneration
 from transformers import TextIteratorStreamer
 from transformers import LlavaNextForConditionalGeneration, LlavaNextProcessor
 from PIL import Image
-import requests
 
 import spaces
 
@@ -65,9 +62,6 @@ def bot_streaming(message, history):
 
     buffer = ""
     for new_text in streamer:
-        # find <|eot_id|> and remove it from the new_text
-        if "<|eot_id|>" in new_text:
-            new_text = new_text.split("<|eot_id|>")[0]
         buffer += new_text
         yield buffer
 
@@ -80,7 +74,7 @@ with gr.Blocks(fill_height=True, ) as demo:
     title="FalconVLM",
     examples=[{"text": "What is on the flower?", "files": ["./bee.jpg"]},
               {"text": "How to make this pastry?", "files": ["./baklava.png"]}],
-    description="Try [tiiuae/falcon-11B-VLM](https://huggingface.co/tiiuae/falcon-11B-vlm). Upload an image and start chatting about it, or simply try one of the examples below. If you don't upload an image, you will receive an error.",
+    description="Try [tiiuae/falcon-11B-VLM](https://huggingface.co/tiiuae/falcon-11B-vlm). Upload an image and start chatting about it, or simply try one of the examples below. If you don't upload an image, you will receive an error. This is not the official demo.",
     stop_btn="Stop Generation",
     multimodal=True,
     textbox=chat_input,