thanhcong2001 commited on
Commit
ae02913
·
verified ·
1 Parent(s): eb6b6e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
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():