Update app.py

app.py

@@ -174,11 +174,6 @@ def save_image(img: Image.Image) -> str:
     img.save(unique_name)
     return unique_name
 
-# Helper to generate a loading animation component.
-def loading_animation(message="Processing..."):
-    # Using Markdown to display an animated GIF for the loading animation.
-    return gr.Markdown(f"![Loading](https://media.giphy.com/media/sSgvbe1m3n93G/giphy.gif) {message}")
-
 @spaces.GPU
 def generate(
     input_dict: dict,
@@ -239,12 +234,9 @@ def generate(
             torch.cuda.empty_cache()
         
         selected_pipe = models.get(model_choice, pipe)
-        
-        # Yield a beautiful loading animation for image generation.
-        yield loading_animation("Generating image...")
-        
         images = selected_pipe(**options).images
         image_path = save_image(images[0])
+        yield "Generating image ..."
         yield gr.Image(image_path)
         return
 
@@ -276,14 +268,11 @@ def generate(
         inputs = processor(text=[prompt], images=images, return_tensors="pt", padding=True).to("cuda")
         streamer = TextIteratorStreamer(processor, skip_prompt=True, skip_special_tokens=True)
         generation_kwargs = {**inputs, "streamer": streamer, "max_new_tokens": max_new_tokens}
-        
-        # Yield a beautiful loading animation for multimodal generation.
-        yield loading_animation("Thinking...")
-        
         thread = Thread(target=model_m.generate, kwargs=generation_kwargs)
         thread.start()
 
         buffer = ""
+        yield "Thinking..."
         for new_text in streamer:
             buffer += new_text
             buffer = buffer.replace("<|im_end|>", "")
@@ -307,10 +296,6 @@ def generate(
             "num_beams": 1,
             "repetition_penalty": repetition_penalty,
         }
-        
-        # Yield a beautiful loading animation for text generation.
-        yield loading_animation("Thinking...")
-        
         t = Thread(target=model.generate, kwargs=generation_kwargs)
         t.start()
 
@@ -342,21 +327,6 @@ h1 {
   background: #1565c0;
   border-radius: 100vh;
 }
-
-/* Loading spinner style (if you prefer a CSS spinner instead) */
-.spinner {
-  margin: 50px auto;
-  width: 50px;
-  height: 50px;
-  border: 6px solid #ddd;
-  border-top: 6px solid #1565c0;
-  border-radius: 50%;
-  animation: spin 1s linear infinite;
-}
-@keyframes spin {
-  0% { transform: rotate(0deg); }
-  100% { transform: rotate(360deg); }
-}
 '''
 
 demo = gr.ChatInterface(
@@ -370,9 +340,9 @@ demo = gr.ChatInterface(
     ],
     examples=[
         ["@tts1 Who is Nikola Tesla, and why did he die?"],
-        ['@lightningv5 Chocolate dripping from a donut against a yellow background, in the style of brocore, hyper-realistic'],
-        ['@lightningv4 " serene landscape with mountains'],
-        ['@turbov3 Abstract art, colorful and vibrant'],
+        ['@lightningv5 "Chocolate dripping from a donut against a yellow background, in the style of brocore, hyper-realistic"'],
+        ['@lightningv4 "A serene landscape with mountains"'],
+        ['@turbov3 "Abstract art, colorful and vibrant"'],
         ["Write a Python function to check if a number is prime."],
         ["@tts2 What causes rainbows to form?"],
     ],