yilmazmusa commited on
Commit
c7f829d
1 Parent(s): 2fe9087

fixed app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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)