Spaces:
Sleeping
Sleeping
Commit
·
856e4ad
1
Parent(s):
65e216d
Update app.py
Browse files
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"
|