Update app.py

app.py

@@ -41,26 +41,36 @@ with open(network_pkl, 'rb') as f:
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 G.eval()
 G.to(device)
-def predict(Seed,choices):
+def predict(Seed,choices,choices2):
 
   shuffle_seed=None
-  w_frames=30*4
   kind='cubic' 
   num_keyframes=None
   wraps=2
   psi=1 
+  s1=Seed
 
   
   if choices=='4x2':
     grid_w = 4
     grid_h = 2
-    s1=Seed
-    seeds=(np.arange(s1-8,s1)).tolist()
+    if choices2=="Large Video":
+        seeds=(np.arange(s1-16,s1)).tolist()  
+        w_frames=60*4
+    if choices2=="Small Video":
+        seeds=(np.arange(s1-8,s1)).tolist()  
+        w_frames=30*4
+        
   if choices=='2x1':
     grid_w = 2
     grid_h = 1
-    s1=Seed
-    seeds=(np.arange(s1-2,s1)).tolist()
+    if choices2=="Large Video":
+        seeds=(np.arange(s1-4,s1)).tolist()  
+        w_frames=60*4
+    if choices2=="Small Video":
+        seeds=(np.arange(s1-2,s1)).tolist()  
+        w_frames=30*4
+
 
 
   mp4='ex.mp4'
@@ -114,10 +124,14 @@ def predict(Seed,choices):
 
 
 choices=['4x2','2x1']
+choices2=["Large Video","Small Video"]
+
 interface=gr.Interface(fn=predict, title="Brain MR Image Generation with StyleGAN-2",
                        description = "",
                        article = "Author: S.Serdar Helli",
-                       inputs=[gr.inputs.Slider( minimum=16, maximum=2**10,label='Seed'),gr.inputs.Radio( choices=choices,  default='4x2',label='Image Grid')],
+                       inputs=[gr.inputs.Slider( minimum=16, maximum=2**10,label='Seed'),gr.inputs.Radio( choices=choices,  default='4x2',label='Image Grid'),
+                       ,gr.inputs.Radio( choices=choices2,  default="Small Video",label='Video Size - It depends on usage of cuda')]],
+                       
                        outputs=gr.outputs.Video(label='Video'),
                        live=False)