Update app.py

app.py

@@ -7,7 +7,7 @@ import time
 import random
 from PIL import Image
 
-m_pdir="//content//drive//MyDrive//aipic001//"
+m_pdir="/content/drive/MyDrive/aipic001/"
 
 models =[
     "",
@@ -172,14 +172,23 @@ demo = gr.Interface(
     
 )
 #demo.load(update_state_info, inputs=state_info, outputs=state_info, every=0.5, show_progress=False)
-img = Image.open("nsfw.PNG")
-img=img.convert("RGB")
-#img.show() # 이미지 확인
-img.save('memia001_save_01.jpg',"JPEG") # 현재 환경에 memi_save라는 이름으로 저장
-img.save(m_pdir+'memima001_save_01.jpg',"JPEG") # 현재 환경에 memi_save라는 이름으로 저장
 
-#image_mod(img)
 
+def saveTest():
+    try:
+        img = Image.open("nsfw.PNG")
+        img=img.convert("RGB")
+        #img.show() # 이미지 확인
+        img.save('memib001_save_01.jpg',"JPEG") # 현재 환경에 memi_save라는 이름으로 저장
+        img.save(m_pdir+'memibb001_save_01.jpg',"JPEG") # 현재 환경에 memi_save라는 이름으로 저장
+
+        return True
+    except:
+        return False
+        
+
+#image_mod(img)
+saveTest()
 
 def is_google_colab():
     try: