Harsimran19 commited on
Commit
6fbbb75
·
1 Parent(s): dc2e757

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,8 +10,8 @@ gen,transform_gen=gen_model()
10
  # print(gen)
11
  to_img=T.ToPILImage()
12
  # examples=["examples/input_0.png","examples/input_9.png"]
13
- # example_list = [["examples/" + example] for example in os.listdir("examples")]
14
- example_list=['1.jpg','2.jpg']
15
 
16
  # Predict Function
17
  def predict(img):
 
10
  # print(gen)
11
  to_img=T.ToPILImage()
12
  # examples=["examples/input_0.png","examples/input_9.png"]
13
+ example_list = [["examples/" + example] for example in os.listdir("examples")]
14
+ # example_list=['1.jpg','2.jpg']
15
 
16
  # Predict Function
17
  def predict(img):