mirror of
https://github.com/corca-ai/EVAL
synced 2024-10-30 09:20:44 +00:00
fix: add image handler exception (#27)
This commit is contained in:
parent
3b5e7f4872
commit
51e351904a
@ -86,6 +86,17 @@ class FileHandler:
|
|||||||
shutil.copy(src, dst)
|
shutil.copy(src, dst)
|
||||||
else:
|
else:
|
||||||
local_filename = self.download(url)
|
local_filename = self.download(url)
|
||||||
return self.handlers[FileType.from_url(url)].handle(local_filename)
|
|
||||||
|
try:
|
||||||
|
handler = self.handlers[FileType.from_url(url)]
|
||||||
|
except KeyError:
|
||||||
|
if FileType.from_url(url) == FileType.IMAGE:
|
||||||
|
raise Exception(
|
||||||
|
f"No handler for {FileType.from_url(url)}. "
|
||||||
|
f"Please set USE_GPU to True in env/settings.py"
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
raise Exception(f"No handler for {FileType.from_url(url)}")
|
||||||
|
handler.handle(local_filename)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
raise e
|
raise e
|
||||||
|
Loading…
Reference in New Issue
Block a user