Update app.py

app.py

@@ -91,8 +91,8 @@ def progress_bar_html(label: str) -> str:
     return f'''
 <div style="display: flex; align-items: center;">
     <span style="margin-right: 10px; font-size: 14px;">{label}</span>
-    <div style="width: 110px; height: 5px; background-color: #FFF0F5; border-radius: 2px; overflow: hidden;">
-        <div style="width: 100%; height: 100%; background-color: #FF69B4; animation: loading 1.5s linear infinite;"></div>
+    <div style="width: 110px; height: 5px; background-color: #DC143C; border-radius: 2px; overflow: hidden;">
+        <div style="width: 100%; height: 100%; background-color: #FF0000; animation: loading 1.5s linear infinite;"></div>
     </div>
 </div>
 <style>
@@ -104,7 +104,7 @@ def progress_bar_html(label: str) -> str:
     '''
 
 
-@spaces.GPU(duration=60, enable_queue=True)
+@spaces.GPU
 def generate(input_dict: dict, chat_history: list[dict],
              max_new_tokens: int = 1024,
              temperature: float = 0.6,
@@ -160,7 +160,7 @@ def generate(input_dict: dict, chat_history: list[dict],
         thread = Thread(target=model_m.generate, kwargs=generation_kwargs)
         thread.start()
         buffer = ""
-        yield progress_bar_html("Processing video with Qwen2VL")
+        yield progress_bar_html("Processing video with Callisto OCR3")
         for new_text in streamer:
             buffer += new_text
             buffer = buffer.replace("<|im_end|>", "")
@@ -190,7 +190,7 @@ def generate(input_dict: dict, chat_history: list[dict],
         thread = Thread(target=model_m.generate, kwargs=generation_kwargs)
         thread.start()
         buffer = ""
-        yield progress_bar_html("Thinking...")
+        yield progress_bar_html("Processing image with Callisto OCR3")
         for new_text in streamer:
             buffer += new_text
             buffer = buffer.replace("<|im_end|>", "")
@@ -219,7 +219,7 @@ def generate(input_dict: dict, chat_history: list[dict],
         t = Thread(target=model.generate, kwargs=generation_kwargs)
         t.start()
         outputs = []
-        yield progress_bar_html("Processing...")
+        yield progress_bar_html("Processing With Pocket Llama 3B")
         for new_text in streamer:
             outputs.append(new_text)
             yield "".join(outputs)