2024.07.04 change appropriate app.py

.ipynb_checkpoints/app-checkpoint.py

@@ -8,8 +8,8 @@ import gradio as gr
 # 모델 로드
 processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
 model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50")
-pipe = StableDiffusionInpaintPipeline.from_pretrained("runwayml/stable-diffusion-inpainting", torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+pipe = StableDiffusionInpaintPipeline.from_pretrained("runwayml/stable-diffusion-inpainting", torch_dtype=torch.float32)
+pipe = pipe.to("cpu")
 
 def detect_objects(image):
     # 객체 검출

app.py

@@ -8,8 +8,8 @@ import gradio as gr
 # 모델 로드
 processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
 model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50")
-pipe = StableDiffusionInpaintPipeline.from_pretrained("runwayml/stable-diffusion-inpainting", torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+pipe = StableDiffusionInpaintPipeline.from_pretrained("runwayml/stable-diffusion-inpainting", torch_dtype=torch.float32)
+pipe = pipe.to("cpu")
 
 def detect_objects(image):
     # 객체 검출