yongyeol commited on
Commit
87e6f23
ยท
verified ยท
1 Parent(s): e5341b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -2
app.py CHANGED
@@ -1,5 +1,15 @@
1
- import os
2
- os.environ["HF_FORCE_SAFE_SERIALIZATION"] = "1" # ์ด๋ฏธ ์•ˆ์ „๋ชจ๋“œ # โœ… safetensors๋งŒ ์‚ฌ์šฉํ•˜๋„๋ก ๊ฐ•์ œ
 
 
 
 
 
 
 
 
 
 
3
 
4
  import gradio as gr
5
  from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoTokenizer
 
1
+ import os, subprocess, sys
2
+ os.environ["HF_FORCE_SAFE_SERIALIZATION"] = "1"
3
+
4
+ # โ”€โ”€โ”€ audiocraft ๋™์  ์„ค์น˜ (ํ•œ ๋ฒˆ๋งŒ) โ”€โ”€โ”€
5
+ try:
6
+ from audiocraft.models import MusicGen
7
+ except ModuleNotFoundError:
8
+ subprocess.check_call(
9
+ [sys.executable, "-m", "pip", "install",
10
+ "git+https://github.com/facebookresearch/audiocraft@main", "--no-deps"]
11
+ )
12
+ from audiocraft.models import MusicGen
13
 
14
  import gradio as gr
15
  from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoTokenizer