SerdarHelli commited on
Commit
188ff28
·
1 Parent(s): 72f9405

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -78,8 +78,8 @@ def predict(seed,trunc_psi):
78
  return x,y,z,i,j,k
79
 
80
  def generate(seed,model_name,trunc_psi):
81
-
82
- change_model(models[model_name])
83
  x,y,z,i,j,k=predict(seed,trunc_psi)
84
 
85
 
 
78
  return x,y,z,i,j,k
79
 
80
  def generate(seed,model_name,trunc_psi):
81
+ if model_name:
82
+ change_model(models[model_name])
83
  x,y,z,i,j,k=predict(seed,trunc_psi)
84
 
85