Ruurd commited on
Commit
65d5ebe
·
1 Parent(s): df88b68

Force install transformers

Browse files
Files changed (1) hide show
  1. app.py +9 -0
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