yongyeol commited on
Commit
c3cf7db
Β·
verified Β·
1 Parent(s): 0fa7d33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,10 +1,12 @@
 
 
 
1
  import gradio as gr
2
  from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoTokenizer
3
  from audiocraft.models import MusicGen
4
  from audiocraft.data.audio import audio_write
5
  from PIL import Image
6
  import torch
7
- import os
8
  import tempfile
9
 
10
  # ───── 이미지 캑셔닝 λͺ¨λΈ λ‘œλ”© ─────
 
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
6
  from audiocraft.models import MusicGen
7
  from audiocraft.data.audio import audio_write
8
  from PIL import Image
9
  import torch
 
10
  import tempfile
11
 
12
  # ───── 이미지 캑셔닝 λͺ¨λΈ λ‘œλ”© ─────