hasibzunair commited on
Commit
4616476
·
1 Parent(s): 2bc0273
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,7 +1,9 @@
1
  import os
 
2
  os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
3
  import codecs
4
  import keras.backend.tensorflow_backend as tb
 
5
  tb._SYMBOLIC_SCOPE.value = True
6
  import numpy as np
7
  import gradio as gr
@@ -10,7 +12,7 @@ from PIL import Image
10
  from tensorflow.keras.models import load_model
11
 
12
  # Get model weights
13
- os.system("wget https://huggingface.co/hasibzunair/melanet/blob/main/MelaNet.h5")
14
 
15
  # Load model
16
  model = None
 
1
  import os
2
+
3
  os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
4
  import codecs
5
  import keras.backend.tensorflow_backend as tb
6
+
7
  tb._SYMBOLIC_SCOPE.value = True
8
  import numpy as np
9
  import gradio as gr
 
12
  from tensorflow.keras.models import load_model
13
 
14
  # Get model weights
15
+ os.system("wget https://huggingface.co/hasibzunair/melanet/resolve/main/MelaNet.h5")
16
 
17
  # Load model
18
  model = None