Spaces:
Sleeping
Sleeping
Commit
·
c7f829d
1
Parent(s):
2fe9087
fixed app.py
Browse files
app.py
CHANGED
@@ -47,8 +47,8 @@ def summarize_and_convert_to_audio(pdf_file):
|
|
47 |
print(device)
|
48 |
|
49 |
# Move models and related tensors to CUDA device if available
|
50 |
-
processor = AutoProcessor.from_pretrained("suno/bark")
|
51 |
-
model = AutoModel.from_pretrained("suno/bark").to(device)
|
52 |
|
53 |
# Extract abstract
|
54 |
abstract_text = extract_abstract(pdf_file)
|
|
|
47 |
print(device)
|
48 |
|
49 |
# Move models and related tensors to CUDA device if available
|
50 |
+
processor = AutoProcessor.from_pretrained("suno/bark-small")
|
51 |
+
model = AutoModel.from_pretrained("suno/bark-small").to(device)
|
52 |
|
53 |
# Extract abstract
|
54 |
abstract_text = extract_abstract(pdf_file)
|