amaralibey commited on
Commit
624f840
·
verified ·
1 Parent(s): 2a7a19d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,8 +16,8 @@ import faiss
16
  from transformers import AutoTokenizer
17
  import gradio as gr
18
 
19
- from src.models import TextEncoder
20
- from deployment.load_album import AlbumDataset
21
 
22
  class ImageSearchEngine:
23
  def __init__(
 
16
  from transformers import AutoTokenizer
17
  import gradio as gr
18
 
19
+ from text_encoder import TextEncoder
20
+ from load_album import AlbumDataset
21
 
22
  class ImageSearchEngine:
23
  def __init__(