Luis J Camargo commited on
Commit
4e7deef
·
1 Parent(s): 9b7b4e8

import fix

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import os
3
  import gradio as gr
4
  import torch
5
  import numpy as np
6
- from transformers import WhisperProcessor, AutoConfig, AutoModel, WhisperConfig
7
  from transformers.models.whisper.modeling_whisper import WhisperEncoder
8
  import torch.nn as nn
9
  from safetensors.torch import load_file
 
3
  import gradio as gr
4
  import torch
5
  import numpy as np
6
+ from transformers import WhisperProcessor, AutoConfig, AutoModel, WhisperConfig, WhisperPreTrainedModel
7
  from transformers.models.whisper.modeling_whisper import WhisperEncoder
8
  import torch.nn as nn
9
  from safetensors.torch import load_file