Update app.py

app.py

@@ -1,14 +1,14 @@
-from cProfile import label
-from distutils.log import debug
-from random import choices
-import gradio as gr
+# -*- coding: utf-8 -*-
+"""
+Created on Mon Jan 30 21:24:16 2023
+
+@author: lizhenzhu
+"""
 import re
+import gradio as gr
 import requests
 import os
 import shutil
-import json
-import time
-
 
 try: os.mkdir('images')
 except: print('images dir might already exist...')
@@ -21,7 +21,9 @@ def download_local(url):
     #filename = 'images/{}_{}'.format(int(time.time()), filename)
     print(filename)
     with open(filename, 'wb') as handler:
-        handler.write(resp.content)
+        for chunk in resp.iter_content(chunk_size = 1024): # 1024 bytes
+            if chunk:
+                handler.write(chunk)
     return filename
 
 
@@ -47,7 +49,7 @@ def download(text):
 
     #zipping 
     print('zipping ...')
-    shutil.make_archive('images', 'zip')    
+    shutil.make_archive('images', 'zip','images')    
     
     result =  '{} link found, {}'.format(len(urls), " ,\n ".join(err_msgs))
 
@@ -57,4 +59,4 @@ def download(text):
 iface = gr.Interface(fn=download, 
                     inputs=["text"], 
                     outputs=["file", "text"])
-iface.launch(debug=True)
+iface.launch()