Update load_rock.py

load_rock.py

@@ -2,6 +2,8 @@ from app_vars import *
 from bs4 import BeautifulSoup as bs
 import random as r
 import requests
+from PIL import Image
+from io import BytesIO
 headers = { 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3',
             "Accept" : "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7",
             "Accept-Encoding": "gzip, deflate, br",
@@ -59,4 +61,4 @@ def get_rock_name_url():
         img = requests.get(url,headers = headers).content
         print(img)
         f.write(img)
-    return name, path
+    return name, Image.open(BytesIO(img))