Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ pipe = pipeline(
|
|
29 |
pipe.model.config.forced_decoder_ids = pipe.tokenizer.get_decoder_prompt_ids(language=language, task="transcribe")
|
30 |
|
31 |
# Load quality rating model
|
32 |
-
rating_pipe = pipeline("text-classification", model="
|
33 |
|
34 |
def is_similar(a, b, threshold=0.8):
|
35 |
return SequenceMatcher(None, a, b).ratio() > threshold
|
|
|
29 |
pipe.model.config.forced_decoder_ids = pipe.tokenizer.get_decoder_prompt_ids(language=language, task="transcribe")
|
30 |
|
31 |
# Load quality rating model
|
32 |
+
rating_pipe = pipeline("text-classification", model="Leo0129/CustomModel-multilingual-sentiment-analysis", device=device)
|
33 |
|
34 |
def is_similar(a, b, threshold=0.8):
|
35 |
return SequenceMatcher(None, a, b).ratio() > threshold
|