Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,6 @@ import gradio as gr
|
|
| 8 |
from facenet_pytorch import InceptionResnetV1, MTCNN
|
| 9 |
from datasets import load_dataset
|
| 10 |
from tqdm import tqdm
|
| 11 |
-
import nest_asyncio
|
| 12 |
import wikipedia
|
| 13 |
dataset = load_dataset("tonyassi/celebrity-1000", split="train")
|
| 14 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
|
@@ -97,7 +96,6 @@ def gradio_find(user_img):
|
|
| 97 |
result_imgs.append((img, caption))
|
| 98 |
wiki_text = get_wikipedia_info(name)
|
| 99 |
return result_imgs, wiki_text
|
| 100 |
-
nest_asyncio.apply()
|
| 101 |
with gr.Blocks() as demo:
|
| 102 |
gr.Markdown("## Which Celebrity Do You Look Like?")
|
| 103 |
with gr.Row():
|
|
|
|
| 8 |
from facenet_pytorch import InceptionResnetV1, MTCNN
|
| 9 |
from datasets import load_dataset
|
| 10 |
from tqdm import tqdm
|
|
|
|
| 11 |
import wikipedia
|
| 12 |
dataset = load_dataset("tonyassi/celebrity-1000", split="train")
|
| 13 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
|
|
|
| 96 |
result_imgs.append((img, caption))
|
| 97 |
wiki_text = get_wikipedia_info(name)
|
| 98 |
return result_imgs, wiki_text
|
|
|
|
| 99 |
with gr.Blocks() as demo:
|
| 100 |
gr.Markdown("## Which Celebrity Do You Look Like?")
|
| 101 |
with gr.Row():
|