diff options
-rw-r--r-- | server.py | 22 |
1 files changed, 10 insertions, 12 deletions
@@ -98,18 +98,16 @@ def index(path, parent): # Modified or not in emb emb = None type = mimetypes.guess_type(path)[0] - try: - if type is None and os.path.getsize(path) < 2**16: - with open(path) as f: - emb = model.embed_text(f.read()) - elif type.startswith("audio"): - emb = model.embed_audio(path) - elif type.startswith("image"): - emb = model.embed_image(path) - elif type.startswith("video") and os.path.getsize(path) < 2**25: - emb = model.embed_video(path) - except: - print(traceback.format_exc()) + if isinstance(type, str): + try: + if type.startswith("audio"): + emb = model.embed_audio(path) + elif type.startswith("image"): + emb = model.embed_image(path) + elif type.startswith("video") and os.path.getsize(path) < 2**25: + emb = model.embed_video(path) + except: + print(traceback.format_exc()) if emb is None: # Might be in index but no longer valid |