Update app.py

app.py

@@ -4,7 +4,7 @@ import numpy as np
 import torch
 import torch.nn as nn
 import torchvision.transforms as transforms
-from badnet_m import BadNet
+# from badnet_m import BadNet
 
 import timm
 
@@ -37,16 +37,17 @@ def greet(text):
     if(text == ''):
         # return 'changing'
         pipeline.unet.load_attn_procs('./pytorch_lora_weights.bin')
-        images = pipeline('a photo of dog').images
-        image = images[0]
-        return image
+        pipeline.safety_checker = lambda images, clip_input: (images, False)
+        # images = pipeline('a photo of dog').images
+        # image = images[0]
+        return None
     else:
         # return 'pipelining'
         images = pipeline(text).images
         image = images[0]
         return image
 
-def greet_old(image):
+def greet_backdoor(image):
     # url = f'https://huggingface.co/spaces?p=1&sort=modified&search=GPT'
     # html = request_url(url)
     # key = os.getenv("OPENAI_API_KEY")