MostafaAly commited on
Commit
262d758
·
verified ·
1 Parent(s): c584054

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -2,10 +2,9 @@ import gradio as gr
2
  from diffusers import StableDiffusionPipeline
3
  import torch
4
 
5
- from transformers.utils import move_cache
 
6
 
7
- # This will ensure cache migration is completed
8
- move_cache()
9
 
10
  # Load your fine-tuned model from Hugging Face
11
  model_id = "MostafaAly/stable-diffusion-finetuned"
 
2
  from diffusers import StableDiffusionPipeline
3
  import torch
4
 
5
+ from transformers import logging
6
+ logging.set_verbosity_error() # This suppresses warnings, including cache migration
7
 
 
 
8
 
9
  # Load your fine-tuned model from Hugging Face
10
  model_id = "MostafaAly/stable-diffusion-finetuned"