Update app.py

app.py

@@ -3,7 +3,6 @@ import requests
 import os
 import io
 from PIL import Image
-from gradio_client import Client
 
 # Get the API token from environment variable
 API_TOKEN = os.environ.get("HF_API_TOKEN")
@@ -24,39 +23,19 @@ def generate_image(prompt):
     image_response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
     image_bytes = image_response.content
     image = Image.open(io.BytesIO(image_bytes))
-
-    with st.spinner("Resizing Image..."):
-        client_resize = Client("doevent/AnimateLCM-SVD")
-        resized_image = client_resize.predict(image, api_name="/resize_image")
-    
-    with st.spinner("Animating Image..."):
-        client_animate = Client("doevent/AnimateLCM-SVD")
-        result = client_animate.predict(
-            resized_image,
-            0,
-            True,
-            1,
-            5,
-            1,
-            1.5,
-            576,
-            320,
-            20,
-            api_name="/video"
-        )
     
-    return result
+    return image
 
 # Streamlit interface for user inputs and displaying outputs
 st.title("Morpheus - Dreams Generator")
-st.write("Enter your feelings and moments of the day to generate a summarization along with an AI-generated animated image!")
+st.write("Enter your feelings and moments of the day to generate a summarization along with an AI-generated image!")
 
 inputs = st.text_area("Enter your emotions, expressions, best and worst moments of the day:", height=200, value="Today was a mix of emotions. I felt happy in the morning but sad in the evening. The best moment was meeting a friend, and the worst was a stressful meeting.")
 
-if st.button("Generate Summary and Animated Image"):
+if st.button("Generate Summary and Image"):
     summary = generate_text_summary(inputs)
     st.write("Summary:", summary)
     
     with st.spinner("Generating Image..."):
-        result = generate_image(inputs)
-        st.markdown(f"")
+        image = generate_image(inputs)
+        st.image(image, caption="Generated Image", use_column_width=True)