Update app.py

app.py

@@ -1,5 +1,6 @@
 import torch
 import torchaudio
+import gradio as gr
 from transformers import Wav2Vec2FeatureExtractor,AutoConfig,pipeline
 
 config = AutoConfig.from_pretrained("SeyedAli/Persian-Speech-Emotion-HuBert-V1")
@@ -13,7 +14,7 @@ def speech_file_to_array_fn(path, sampling_rate):
 
 def predict(path, sampling_rate):
     speech = speech_file_to_array_fn(path, sampling_rate)
-    inputs = feature_extractor(speech, sampling_rate=sampling_rate, return_tensors="pt", padding=True)
+    inputs = model(speech, sampling_rate=sampling_rate, return_tensors="pt", padding=True)
     inputs = {key: inputs[key].to(device) for key in inputs}
 
     with torch.no_grad():