Hugging Face's logo

Spaces:
ahmedbrs
/
scene-sketch-seg
Running

App Files Community
ahmedbrs commited on
Commit
39d3b58
·
1 Parent(s): 53e6d40

add multi-class

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. output.png +0 -0
app.py CHANGED
@@ -59,7 +59,7 @@ print("Model loaded successfully")
59
 
60
  def run(sketch, caption, threshold, seed):
61
  # select a random seed between 1 and 10 for the color
62
- color_seed = np.random.randint(0, 9)
63
 
64
  # set the condidate classes here
65
  caption = caption.replace('\n',' ')
 
59
 
60
  def run(sketch, caption, threshold, seed):
61
  # select a random seed between 1 and 10 for the color
62
+ color_seed = np.random.randint(0, 4)
63
 
64
  # set the condidate classes here
65
  caption = caption.replace('\n',' ')
output.png CHANGED