asrs777 commited on
Commit
c3b2098
1 Parent(s): c4fc218

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,7 +9,7 @@ from PIL import Image
9
  import itertools
10
  device = 1 if torch.cuda.is_available() else "cpu"
11
 
12
- chk_point = "./autotrain-diffusion-emotion-facial-expression-recognition-40429105176"
13
 
14
  model = AutoModelForImageClassification.from_pretrained(chk_point)
15
 
@@ -148,4 +148,4 @@ gr.Interface(
148
  outputs=["image", "label"],
149
  examples=[["examples/9_peoples.jpg"], ["examples/sad.jpg"], ["examples/angry.jpg"], ["examples/surprise.jpg"]],
150
  title="FER trained on DiffusionFER dataset...",
151
- ).launch(share=True)
 
9
  import itertools
10
  device = 1 if torch.cuda.is_available() else "cpu"
11
 
12
+ chk_point = "kdhht2334/autotrain-diffusion-emotion-facial-expression-recognition-40429105176"
13
 
14
  model = AutoModelForImageClassification.from_pretrained(chk_point)
15
 
 
148
  outputs=["image", "label"],
149
  examples=[["examples/9_peoples.jpg"], ["examples/sad.jpg"], ["examples/angry.jpg"], ["examples/surprise.jpg"]],
150
  title="FER trained on DiffusionFER dataset...",
151
+ ).launch()