Update app.py

app.py

@@ -7,6 +7,8 @@ import time
 import random
 from PIL import Image
 
+m_pdir="//content//drive//Mydrive//aipic001//"
+
 models =[
     "",
     "CompVis/stable-diffusion-v1-4",
@@ -142,7 +144,7 @@ def image_mod(image):
     #img = Image.open(output1) # 이미지 파일명 또는 파일 위치
     #img=img.convert("RGB")
     #img.save('save_'+ppStr+'.jpg',"JPEG")
-    pdir="//content//drive//Mydrive//aipic001//"
+    
     saveImg(output1,'save_1_'+ppStr+'.jpg')
     saveImg(output2,'save_2_'+ppStr+'.jpg')
     saveImg(output3,'save_3_'+ppStr+'.jpg')
@@ -173,7 +175,8 @@ demo = gr.Interface(
 img = Image.open("nsfw.PNG")
 img=img.convert("RGB")
 #img.show() # 이미지 확인
-img.save('memi_save_01.jpg',"JPEG") # 현재 환경에 memi_save라는 이름으로 저장
+img.save('memia001_save_01.jpg',"JPEG") # 현재 환경에 memi_save라는 이름으로 저장
+img.save(m_pdir+'memima001_save_01.jpg',"JPEG") # 현재 환경에 memi_save라는 이름으로 저장
 
 #image_mod(img)