Update app.py

app.py

@@ -77,7 +77,7 @@ def main():
     #else:
     #    ini = "False1"
     #result = subprocess.check_output(['nvidia-smi'])
-    
+    #load_model("stylegan2-ffhq-config-f","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",device)
     args = ProjectorArguments().parse(
         args=[str(input_path)],
         namespace=Namespace(
@@ -88,9 +88,8 @@ def main():
             log_visual_freq=1000,
             input='text',
 ))
-    #device = torch.device(args.device)
-    #load_model("stylegan2-ffhq-config-f","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",device)
-    
+    device = th.device()
+    generator = create_generator(args, device)
     iface = gr.Interface(
       fn=predict, 
       inputs='text',