Do not allow download_tmp_copy_of_file to download error responses as…

… valid files
gradio-app · Jun 27, 2023 · 500dc72 · 500dc72
1 parent 91932b4
commit 500dc72
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/client/python/gradio_client/utils.py b/client/python/gradio_client/utils.py
@@ -304,10 +304,10 @@ def download_tmp_copy_of_file(
     directory.mkdir(exist_ok=True, parents=True)
     file_path = directory / Path(url_path).name
 
-    with requests.get(url_path, headers=headers, stream=True) as r, open(
-        file_path, "wb"
-    ) as f:
-        shutil.copyfileobj(r.raw, f)
+    with requests.get(url_path, headers=headers, stream=True) as r:
+        r.raise_for_status()
+        with open(file_path, "wb") as f:
+            shutil.copyfileobj(r.raw, f)
     return str(file_path.resolve())