Spaces:
Running
Running
Commit
·
6e8a7f9
1
Parent(s):
960b1a0
add spaces.GPU
Browse files- app.py +2 -0
- requirements.txt +0 -0
app.py
CHANGED
|
@@ -13,6 +13,7 @@ from data_loading.feature_extractor import (
|
|
| 13 |
import chardet
|
| 14 |
import torch
|
| 15 |
from models.models import BiFormer
|
|
|
|
| 16 |
|
| 17 |
|
| 18 |
DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
|
@@ -150,6 +151,7 @@ def get_top_emotion(probabilities):
|
|
| 150 |
return f"{LABEL_TO_EMOTION[max_idx]} ({max(probabilities)*100:.1f}%)"
|
| 151 |
|
| 152 |
|
|
|
|
| 153 |
def process_audio(audio_path):
|
| 154 |
"""Main processing pipeline."""
|
| 155 |
try:
|
|
|
|
| 13 |
import chardet
|
| 14 |
import torch
|
| 15 |
from models.models import BiFormer
|
| 16 |
+
import spaces
|
| 17 |
|
| 18 |
|
| 19 |
DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
|
|
|
| 151 |
return f"{LABEL_TO_EMOTION[max_idx]} ({max(probabilities)*100:.1f}%)"
|
| 152 |
|
| 153 |
|
| 154 |
+
@spaces.GPU
|
| 155 |
def process_audio(audio_path):
|
| 156 |
"""Main processing pipeline."""
|
| 157 |
try:
|
requirements.txt
CHANGED
|
Binary files a/requirements.txt and b/requirements.txt differ
|
|
|