Andrei Boiarov commited on
Commit
688eb37
1 Parent(s): 4d7625a

Update app file

Browse files
Files changed (2) hide show
  1. 1.jpeg +0 -0
  2. app.py +5 -1
1.jpeg DELETED
Binary file (125 kB)
 
app.py CHANGED
@@ -2,6 +2,7 @@ from transformers import ViTFeatureExtractor, ViTMAEForPreTraining
2
  import numpy as np
3
  import torch
4
  from PIL import Image
 
5
 
6
  import gradio as gr
7
 
@@ -45,7 +46,10 @@ def reconstruct(img):
45
  # # out_masked, out_rec, out_rec_vis = img, img, img
46
  #
47
  # return [(out_masked, 'masked'), (out_rec, 'reconstruction'), (out_rec_vis, 'reconstruction + visible')]
48
- return [("https://huggingface.co/spaces/andrewbo29/f1-mae-reconstruction/blob/main/1.jpeg", 'label 1')]
 
 
 
49
 
50
 
51
  with gr.Blocks() as demo:
 
2
  import numpy as np
3
  import torch
4
  from PIL import Image
5
+ import requests
6
 
7
  import gradio as gr
8
 
 
46
  # # out_masked, out_rec, out_rec_vis = img, img, img
47
  #
48
  # return [(out_masked, 'masked'), (out_rec, 'reconstruction'), (out_rec_vis, 'reconstruction + visible')]
49
+ url = "https://images.unsplash.com/photo-1507003211169-0a1dd7228f2d?ixlib=rb-1.2.1&ixid=MnwxMjA3fDB8MHxwaG90by1wYWdlfHx8fGVufDB8fHx8&auto=format&fit=crop&w=387&q=80"
50
+ im = Image.open(requests.get(url, stream=True).raw)
51
+ im = np.array(im)
52
+ return [(im, 'label 1')]
53
 
54
 
55
  with gr.Blocks() as demo: