Update app.py
Browse files
app.py
CHANGED
@@ -4,6 +4,9 @@ os.system("pip install git+https://github.com/openai/whisper.git")
|
|
4 |
import gradio as gr
|
5 |
import whisper
|
6 |
from transformers import pipeline
|
|
|
|
|
|
|
7 |
# from next_word_prediction import GPT2
|
8 |
|
9 |
from share_btn import community_icon_html, loading_icon_html, share_js
|
|
|
4 |
import gradio as gr
|
5 |
import whisper
|
6 |
from transformers import pipeline
|
7 |
+
import torch
|
8 |
+
from transformers import AutoModelForCausalLM
|
9 |
+
from transformers import AutoTokenizer
|
10 |
# from next_word_prediction import GPT2
|
11 |
|
12 |
from share_btn import community_icon_html, loading_icon_html, share_js
|