shireenchand commited on
Commit
856e4ad
·
1 Parent(s): 65e216d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -8,6 +8,8 @@ classes = ['paper', 'rock', 'scissors']
8
 
9
  model = tf.keras.models.load_model('rps.h5')
10
 
 
 
11
  # sel = None
12
  # det = None
13
 
@@ -16,21 +18,23 @@ def classify_image(inp):
16
  img = np.reshape(img,(1,224,224,3))
17
  pred = model.predict(img).flatten()
18
  confidences = {classes[i]: float(pred[i]) for i in range(3)}
19
- global det
20
  det = classes[pred.argmax(axis=-1)]
21
  print(det)
 
22
  return confidences
23
 
24
  def random_char(n):
25
  print("HELLO")
26
  n = random.randint(0,2)
27
- global sel
28
  sel = classes[n]
29
  print(sel)
 
30
  return classes[n].upper()
31
 
32
  def result(t):
33
  print("HIOAL")
 
 
34
  print(sel, det)
35
  if (sel == 'rock' and det == 'paper') or (sel == 'paper' and det == 'scissors') or (sel == "scissors" and det == "rock"):
36
  return "YOU WON"
 
8
 
9
  model = tf.keras.models.load_model('rps.h5')
10
 
11
+ ch = {}
12
+
13
  # sel = None
14
  # det = None
15
 
 
18
  img = np.reshape(img,(1,224,224,3))
19
  pred = model.predict(img).flatten()
20
  confidences = {classes[i]: float(pred[i]) for i in range(3)}
 
21
  det = classes[pred.argmax(axis=-1)]
22
  print(det)
23
+ ch['det'] = det
24
  return confidences
25
 
26
  def random_char(n):
27
  print("HELLO")
28
  n = random.randint(0,2)
 
29
  sel = classes[n]
30
  print(sel)
31
+ ch['sel'] = sel
32
  return classes[n].upper()
33
 
34
  def result(t):
35
  print("HIOAL")
36
+ sel = ch['sel']
37
+ det = ch['det']
38
  print(sel, det)
39
  if (sel == 'rock' and det == 'paper') or (sel == 'paper' and det == 'scissors') or (sel == "scissors" and det == "rock"):
40
  return "YOU WON"