dongsheng commited on
Commit
865f10a
·
1 Parent(s): 1e947cf

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -38,7 +38,7 @@ def img2label(left, right):
38
 
39
 
40
  if __name__ == '__main__':
41
- device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
42
 
43
  # 标题
44
  title = "基于眼底图像的智能健康诊断分析系统"
@@ -69,5 +69,5 @@ if __name__ == '__main__':
69
  examples = [[left_img_dir, right_img_dir]]
70
  # r = img2label(left_img_dir, right_img_dir)
71
  demo = gr.Interface(fn=img2label, inputs=[gr.inputs.Image(), gr.inputs.Image()],
72
- outputs="text", title=title, description=description)
73
  demo.launch()
 
38
 
39
 
40
  if __name__ == '__main__':
41
+ device = torch.device("cpu")
42
 
43
  # 标题
44
  title = "基于眼底图像的智能健康诊断分析系统"
 
69
  examples = [[left_img_dir, right_img_dir]]
70
  # r = img2label(left_img_dir, right_img_dir)
71
  demo = gr.Interface(fn=img2label, inputs=[gr.inputs.Image(), gr.inputs.Image()],
72
+ outputs="text", examples=examples, title=title, description=description)
73
  demo.launch()