Update gradio_tabs/animation.py

gradio_tabs/animation.py

@@ -124,7 +124,7 @@ def vid_postprocessing(video, fps, output_path=output_dir + "/output_vid.mp4"):
 
 	return output_path
 
-
+@torch.no_grad()
 def edit_media(image, *selected_s):
 
 	image_tensor = img_preprocessing(image, 512)
@@ -137,7 +137,7 @@ def edit_media(image, *selected_s):
 
 	return edited_image
 
-
+@torch.no_grad()
 def animate_media(image, video, *selected_s):
 
 	image_tensor = img_preprocessing(image, 512)
@@ -161,7 +161,7 @@ def clear_media():
 # video_output = gr.Video(label="Output Video", elem_id="output_vid", width=512)
 
 
-@torch.no_grad()
+
 def animation():
     
 	with gr.Tab("Animation & Image Editing"):