saranbalan commited on
Commit
ce1d9ec
·
verified ·
1 Parent(s): ef96a1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -31,9 +31,9 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
31
  whisper_model = whisper.load_model("base")
32
 
33
  # Model IDs for Stable Diffusion pipelines
34
- model_id1 = "dreamlike-art/dreamlike-diffusion-1.0"
35
- model_id2 = "stabilityai/stable-diffusion-xl-base-1.0"
36
- restricted_model_id = "ByteDance/SDXL-Lightning" # Model to access using HF_API_KEY
37
 
38
  # Function to transcribe, translate, and analyze sentiment
39
  def process_audio(audio_path, image_option):
 
31
  whisper_model = whisper.load_model("base")
32
 
33
  # Model IDs for Stable Diffusion pipelines
34
+ # model_id1 = "dreamlike-art/dreamlike-diffusion-1.0"
35
+ # model_id2 = "stabilityai/stable-diffusion-xl-base-1.0"
36
+ restricted_model_id = "https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_base_1.0.safetensors" # Model to access using HF_API_KEY
37
 
38
  # Function to transcribe, translate, and analyze sentiment
39
  def process_audio(audio_path, image_option):