haoheliu commited on
Commit
958907e
1 Parent(s): c2362ff
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -2,11 +2,8 @@ import gradio as gr
2
  import numpy as np
3
  from audioldm import text_to_audio, build_model
4
  from share_btn import community_icon_html, loading_icon_html, share_js
5
- # from transformers import AutoModelForCausalLM, AutoTokenizer
6
- # import torch
7
 
8
- # tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
9
- # model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
10
 
11
  audioldm = build_model()
12
  # audioldm=None
 
2
  import numpy as np
3
  from audioldm import text_to_audio, build_model
4
  from share_btn import community_icon_html, loading_icon_html, share_js
 
 
5
 
6
+ model_id="haoheliu/AudioLDM-S-Full"
 
7
 
8
  audioldm = build_model()
9
  # audioldm=None