Spaces:

atticus
/

image-text-retrival-huster

Runtime error

atticus commited on Mar 10, 2022

Commit

6921c40

•

1 Parent(s): 1e7fce7

w/o translate

Files changed (1) hide show

app.py CHANGED Viewed

@@ -60,8 +60,8 @@ def download_url_img(url):
 def search(mode, image, text):
-    translator = Translator(from_lang="chinese",to_lang="english")
-    text = translator.translate(text)
     if mode == T2I:
         dataset = torch.Tensor(encoder.encode(text)).unsqueeze(dim=0)
         dataset_loader = DataLoader(dataset, batch_size=batch_size, num_workers=1, pin_memory=True, collate_fn=collate_fn_cap_padded)

 def search(mode, image, text):
+    # translator = Translator(from_lang="chinese",to_lang="english")
+    # text = translator.translate(text)
     if mode == T2I:
         dataset = torch.Tensor(encoder.encode(text)).unsqueeze(dim=0)
         dataset_loader = DataLoader(dataset, batch_size=batch_size, num_workers=1, pin_memory=True, collate_fn=collate_fn_cap_padded)