Hugging Face's logo

Spaces:
keras-io
/
Attention_based_Deep_Multiple_Instance_Learning
Runtime error

App Files Community
1
geninhu commited on
Commit
0a4335a
·
1 Parent(s): c56dc17

Revise mode path

Browse files
Files changed (2) hide show
  1. README.md +2 -2
  2. app.py +1 -1
README.md CHANGED
@@ -1,8 +1,8 @@
1
  ---
2
  title: Attention Based Deep Multiple Instance Learning
3
- emoji: 🦀
4
  colorFrom: yellow
5
- colorTo: red
6
  sdk: gradio
7
  sdk_version: 3.0.13
8
  app_file: app.py
 
1
  ---
2
  title: Attention Based Deep Multiple Instance Learning
3
+ emoji: 💰
4
  colorFrom: yellow
5
+ colorTo: blue
6
  sdk: gradio
7
  sdk_version: 3.0.13
8
  app_file: app.py
app.py CHANGED
@@ -10,7 +10,7 @@ import gradio as gr
10
  from huggingface_hub import from_pretrained_keras
11
 
12
 
13
- model = from_pretrained_keras('geninhu/attention_mil')
14
 
15
  # functions for inference
16
  IMG_SIZE = 28
 
10
  from huggingface_hub import from_pretrained_keras
11
 
12
 
13
+ model = from_pretrained_keras('keras-io/attention_mil')
14
 
15
  # functions for inference
16
  IMG_SIZE = 28