Spaces:
Sleeping
Sleeping
Andrei Boiarov
commited on
Commit
•
688eb37
1
Parent(s):
4d7625a
Update app file
Browse files
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 |
-
|
|
|
|
|
|
|
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:
|