cycool29 commited on
Commit
36a919c
·
1 Parent(s): a1ae495

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -14,22 +14,23 @@ def process_file(webcam_filepath, upload_filepath):
14
  for class_label, class_prob in sorted_classes:
15
  class_prob = class_prob.item().__round__(2)
16
  result.append(f"{class_label}: {class_prob}%")
17
- return sorted_classes
18
  elif upload_filepath == None:
19
  sorted_classes = predict.predict_image(webcam_filepath)
20
  for class_label, class_prob in sorted_classes:
21
  class_prob = class_prob.item().__round__(2)
22
  result.append(f"{class_label}: {class_prob}%")
23
- return sorted_classes
24
  else:
25
  sorted_classes = predict.predict_image(upload_filepath)
26
  for class_label, class_prob in sorted_classes:
27
  class_prob = class_prob.item().__round__(2)
28
  result.append(f"{class_label}: {class_prob}%")
29
- return sorted_classes
30
 
31
 
32
  demo = gr.Interface(
 
33
  fn=process_file,
34
  title="HANDETECT",
35
  description="An innovative AI-powered system that facilitates early detection and monitoring of movement disorders through handwriting assessment",
 
14
  for class_label, class_prob in sorted_classes:
15
  class_prob = class_prob.item().__round__(2)
16
  result.append(f"{class_label}: {class_prob}%")
17
+ return result
18
  elif upload_filepath == None:
19
  sorted_classes = predict.predict_image(webcam_filepath)
20
  for class_label, class_prob in sorted_classes:
21
  class_prob = class_prob.item().__round__(2)
22
  result.append(f"{class_label}: {class_prob}%")
23
+ return result
24
  else:
25
  sorted_classes = predict.predict_image(upload_filepath)
26
  for class_label, class_prob in sorted_classes:
27
  class_prob = class_prob.item().__round__(2)
28
  result.append(f"{class_label}: {class_prob}%")
29
+ return result
30
 
31
 
32
  demo = gr.Interface(
33
+ theme='gradio/soft',
34
  fn=process_file,
35
  title="HANDETECT",
36
  description="An innovative AI-powered system that facilitates early detection and monitoring of movement disorders through handwriting assessment",