Staticaliza commited on
Commit
842b1c3
·
verified ·
1 Parent(s): ca4e7c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -90,13 +90,11 @@ def generate(input, instruction=DEFAULT_INPUT, sampling=False, temperature=0.7,
90
  elif filetype == "Video":
91
  frames = encode_video(input)
92
  content.extend(frames)
93
- audio_np, sample_rate = librosa.load(input, sr=16000, mono=True)
94
- audio_tensor = torch.from_numpy(audio_np).float().to(DEVICE)
95
- content.append({"array": audio_tensor, "sampling_rate": sample_rate})
96
  elif filetype == "Audio":
97
- audio_np, sample_rate = librosa.load(input, sr=16000, mono=True)
98
- audio_tensor = torch.from_numpy(audio_np).float().to(DEVICE)
99
- content.append({"array": audio_tensor, "sampling_rate": sample_rate})
100
  else:
101
  return "Unsupported file type."
102
 
 
90
  elif filetype == "Video":
91
  frames = encode_video(input)
92
  content.extend(frames)
93
+ audio, _ = librosa.load(input, sr=16000, mono=True)
94
+ content.append(audio)
 
95
  elif filetype == "Audio":
96
+ audio, _ = librosa.load(input, sr=16000, mono=True)
97
+ content.append(audio)
 
98
  else:
99
  return "Unsupported file type."
100