Update app.py

app.py

@@ -4,7 +4,7 @@ import torch
 from diffusers import StableDiffusionUpscalePipeline
 
 # 모델 로드
-pipe = StableDiffusionUpscalePipeline.from_pretrained("stabilityai/stable-diffusion-x4-upscaler", torch_dtype=torch.float16)
+pipe = StableDiffusionUpscalePipeline.from_pretrained("stabilityai/stable-diffusion-x4-upscaler")
 
 # GPU가 사용 가능한지 확인한 후, 가능하면 GPU로 설정하고, 그렇지 않으면 CPU로 설정
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -21,8 +21,9 @@ def upscale_image(image, scale_factor):
         raise ValueError("지원하지 않는 배율입니다.")
 
     # 이미지 업스케일링 수행
+    prompt = "high quality, detailed"
     with torch.no_grad():  # 메모리 최적화를 위해 no_grad 사용
-        upscaled_image = pipe(image=low_res_image).images[0]
+        upscaled_image = pipe(prompt=prompt, image=low_res_image).images[0]
     return upscaled_image
 
 # Gradio 인터페이스 설정
@@ -50,4 +51,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-