Force install transformers
Browse files
app.py
CHANGED
@@ -1,8 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
import gradio as gr
|
2 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
3 |
import torch
|
4 |
import spaces
|
5 |
|
|
|
6 |
# Dictionary to store loaded models and tokenizers
|
7 |
loaded_models = {}
|
8 |
|
|
|
1 |
+
import os
|
2 |
+
import subprocess
|
3 |
+
|
4 |
+
def install(package):
|
5 |
+
subprocess.check_call([os.sys.executable, "-m", "pip", "install", package])
|
6 |
+
|
7 |
+
install("transformers")
|
8 |
+
|
9 |
import gradio as gr
|
10 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
11 |
import torch
|
12 |
import spaces
|
13 |
|
14 |
+
|
15 |
# Dictionary to store loaded models and tokenizers
|
16 |
loaded_models = {}
|
17 |
|