cpu;

README.md

@@ -6,6 +6,6 @@ colorTo: indigo
 sdk: gradio
 sdk_version: 3.16.0
 app_file: app.py
-pinned: false
+pinned: true
 license: unknown
 ---

src/main/inference.py

@@ -43,7 +43,7 @@ class Inference():
         self.only_fl = args.only_fl
         self.output_name = args.output_name
         self.test_img2img = args.test_img2img
-        self.seq_len = 5#args.seq_len
+        self.seq_len = 5
         self.model_type = args.model_type
         
         
@@ -140,9 +140,6 @@ class Inference():
                     bug = fls[i]
             
                     fl1 = bug[:68]
-                    
-                    #fl2 = bug[68:]
-                    
                  
                     
                     fls[i] = fl1
@@ -412,7 +409,7 @@ class Inference():
                     
                         with torch.no_grad():  
                             self.image2image.eval() 
-                            print("trans in", trans_in.is_cuda)
+                            
                             trans_out = self.image2image(trans_in)
                             trans_out = torch.tanh(trans_out)