Update main.py

main.py

@@ -1,7 +1,6 @@
 import Linlada
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-from diffusions import StableDiffusionPipeline
 
 app = FastAPI()
 
@@ -13,17 +12,6 @@ app.add_middleware(  # add the middleware
     allow_headers=["*"],  # allow all headers
 )
 
-
-model_id = "runwayml/stable-diffusion-v1-5"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=auth_token)
-pipe = pipe.to("cpu")
-pipe.enable_attention_slicing()
-
-def dummy(images, **kwargs):
-    return images, False
-
-pipe.safety_checker = dummy
-
 @app.get("/")
 def hello():
     return "Hello, I'm Artist"