black44 commited on
Commit
f7a2c0c
·
verified ·
1 Parent(s): 7b8c16b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  from fastapi import FastAPI, Form
2
  from fastapi.responses import FileResponse, JSONResponse
3
  from pydantic import BaseModel
@@ -8,6 +9,12 @@ import uuid
8
  import os
9
  from typing import Optional
10
 
 
 
 
 
 
 
11
  # Load TTS model and processor
12
  processor = AutoProcessor.from_pretrained("suno/bark")
13
  model = BarkModel.from_pretrained("suno/bark")
 
1
+ from transformers import AutoTokenizer, AutoProcessor
2
  from fastapi import FastAPI, Form
3
  from fastapi.responses import FileResponse, JSONResponse
4
  from pydantic import BaseModel
 
9
  import os
10
  from typing import Optional
11
 
12
+ # Load the pre-downloaded model and tokenizer
13
+ tokenizer = AutoTokenizer.from_pretrained("/app/models/suno-bark")
14
+ processor = AutoProcessor.from_pretrained("/app/models/suno-bark")
15
+
16
+ # Rest of your application code...
17
+
18
  # Load TTS model and processor
19
  processor = AutoProcessor.from_pretrained("suno/bark")
20
  model = BarkModel.from_pretrained("suno/bark")