Update app.py

app.py

@@ -45,7 +45,6 @@ def seed_all(seed):
 
 
 def change_model(ckpt_path):
-    global model
     if device=="cuda":
       model = StyleGAN2_3D.load_from_checkpoint(cars).cuda(0)
     else:
@@ -78,10 +77,9 @@ def predict(seed,trunc_psi):
 
   return x,y,z,i,j,k
 
-def generate(seed,model,trunc_psi):
+def generate(seed,model_name,trunc_psi):
 
-    global model
-    change_model(models[model])
+    change_model(models[model_name])
     x,y,z,i,j,k=predict(seed,trunc_psi)
 
 
@@ -123,12 +121,12 @@ with gr.Blocks() as demo:
             gr.Markdown(markdown)
         with gr.Row():
             seed = gr.Slider( minimum=0, maximum=2**16,label='Seed')
-            model=gr.Dropdown(choices=["Car","Airplane","Chair","Rifle","Table"],label="Choose Model Type")
+            model_name=gr.Dropdown(choices=["Car","Airplane","Chair","Rifle","Table"],label="Choose Model Type")
             trunc_psi = gr.Slider( minimum=0, maximum=2,label='Truncate PSI')
 
         btn = gr.Button(value="Generate")
         mesh = gr.Plot()
-    demo.load(generate, [seed,model,trunc_psi], mesh)
-    btn.click(generate, [seed,model,trunc_psi], mesh)
+    demo.load(generate, [seed,model_name,trunc_psi], mesh)
+    btn.click(generate, [seed,model_name,trunc_psi], mesh)
 
 demo.launch(debug=True)