dtype reg. update
Browse files
app.py
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
"""
|
| 2 |
Dialect Intelligence Engine
|
| 3 |
Author: Nisa Çayır
|
| 4 |
-
Core:
|
| 5 |
Focus: Turkish dialects — vowel shifts, markers, prosody, phonetic signatures
|
| 6 |
"""
|
| 7 |
|
|
@@ -59,11 +59,15 @@ logger.info(f"Using device: {DEVICE}, dtype: {DTYPE}")
|
|
| 59 |
# =========================================
|
| 60 |
try:
|
| 61 |
processor = AutoProcessor.from_pretrained(MODEL_ID)
|
|
|
|
| 62 |
model = AutoModelForSpeechSeq2Seq.from_pretrained(
|
| 63 |
MODEL_ID,
|
| 64 |
-
|
| 65 |
-
)
|
|
|
|
|
|
|
| 66 |
model.eval()
|
|
|
|
| 67 |
logger.info("Model loaded successfully")
|
| 68 |
except Exception as e:
|
| 69 |
logger.error(f"Error loading model: {e}")
|
|
@@ -843,4 +847,4 @@ if __name__ == "__main__":
|
|
| 843 |
server_name="0.0.0.0",
|
| 844 |
server_port=7860,
|
| 845 |
share=False
|
| 846 |
-
)
|
|
|
|
| 1 |
"""
|
| 2 |
Dialect Intelligence Engine
|
| 3 |
Author: Nisa Çayır
|
| 4 |
+
Core: Whisper Large-v3 + Türkçe Bölgesel Fonetik Analiz
|
| 5 |
Focus: Turkish dialects — vowel shifts, markers, prosody, phonetic signatures
|
| 6 |
"""
|
| 7 |
|
|
|
|
| 59 |
# =========================================
|
| 60 |
try:
|
| 61 |
processor = AutoProcessor.from_pretrained(MODEL_ID)
|
| 62 |
+
|
| 63 |
model = AutoModelForSpeechSeq2Seq.from_pretrained(
|
| 64 |
MODEL_ID,
|
| 65 |
+
torch_dtype=DTYPE
|
| 66 |
+
)
|
| 67 |
+
|
| 68 |
+
model = model.to(DEVICE)
|
| 69 |
model.eval()
|
| 70 |
+
|
| 71 |
logger.info("Model loaded successfully")
|
| 72 |
except Exception as e:
|
| 73 |
logger.error(f"Error loading model: {e}")
|
|
|
|
| 847 |
server_name="0.0.0.0",
|
| 848 |
server_port=7860,
|
| 849 |
share=False
|
| 850 |
+
)
|