Update app.py

app.py

@@ -4,13 +4,10 @@ import requests
 from PIL import Image
 from io import BytesIO
 
-# Set the replicate API token
-import os
-os.environ["REPLICATE_API_TOKEN"] = "r8_JSR8xlRoCk6cmq3qEOOThVTn3dAgdPq1bWXdj"
-
-def run_replicate_model(input_params):
+def run_replicate_model(api_key, input_params):
     output = replicate.run(
         "stability-ai/stable-video-diffusion:3f0457e4619daac51203dedb472816fd4af51f3149fa7a9e0b5ffcf1b8172438",
+        api_key=api_key,
         input=input_params
     )
     return output
@@ -18,6 +15,9 @@ def run_replicate_model(input_params):
 st.title("Stable Video Diffusion")
 st.write("Fill in the input parameters and click on the button to run the Stable Video Diffusion model.")
 
+# Input field for Replicate API token
+api_key = st.text_input("Replicate API Token")
+
 # Allow users to upload an image file or provide a URL
 uploaded_file = st.file_uploader("Upload Image File", type=["jpg", "png", "jpeg"])
 input_image = None
@@ -46,12 +46,15 @@ input_params = {
 }
 
 if st.button("Run Model"):
-    output_result = run_replicate_model(input_params)
-    st.write("Output Video:")
-    
-    try:
-        response = requests.get(output_result)
-        video_bytes = BytesIO(response.content)
-        st.video(video_bytes)
-    except Exception as e:
-        st.write("Unable to display the video.")
+    if api_key:
+        output_result = run_replicate_model(api_key, input_params)
+        st.write("Output Video:")
+        
+        try:
+            response = requests.get(output_result)
+            video_bytes = BytesIO(response.content)
+            st.video(video_bytes)
+        except Exception as e:
+            st.write("Unable to display the video.")
+    else:
+        st.write("Please provide your Replicate API Token.")