svjack commited on
Commit
b924eac
1 Parent(s): 5e23189

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -73,7 +73,7 @@ def perform_neural_transfer(content_image_input, style_image_input, super_resolu
73
  print(super_resolution_type)
74
  #print(super_resolution_type.value)
75
 
76
- if super_resolution_type is "none":
77
  return stylized_image
78
  else:
79
  print("call else :")
 
73
  print(super_resolution_type)
74
  #print(super_resolution_type.value)
75
 
76
+ if super_resolution_type not in ["base", "anime"]:
77
  return stylized_image
78
  else:
79
  print("call else :")