irena commited on
Commit
cd3acfe
·
1 Parent(s): 240bd36

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -2,10 +2,13 @@ import os
2
  import gradio as gr
3
  from transformers import pipeline
4
  import gradio as gr
 
 
5
 
6
  os.system('pip install https://huggingface.co/Armandoliv/es_pipeline/resolve/main/es_pipeline-any-py3-none-any.whl')
7
 
8
  pipe = pipeline(model="irena/whisper-small-sv-SE")
 
9
  def main_generator(youtube_id:str):
10
  YouTubeID = youtube_id.split("https://www.youtube.com/watch?v=") #
11
  if len(YouTubeID)>1:
 
2
  import gradio as gr
3
  from transformers import pipeline
4
  import gradio as gr
5
+ import torch
6
+ import spacy
7
 
8
  os.system('pip install https://huggingface.co/Armandoliv/es_pipeline/resolve/main/es_pipeline-any-py3-none-any.whl')
9
 
10
  pipe = pipeline(model="irena/whisper-small-sv-SE")
11
+ nlp_ner = spacy.load("es_pipeline")
12
  def main_generator(youtube_id:str):
13
  YouTubeID = youtube_id.split("https://www.youtube.com/watch?v=") #
14
  if len(YouTubeID)>1: