Nemil commited on
Commit
c04a815
Β·
verified Β·
1 Parent(s): 85f688c

Modify app

Browse files
Files changed (1) hide show
  1. app.py +21 -0
app.py CHANGED
@@ -1,3 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  from transformers import AutoProcessor, AutoModelForCausalLM, BitsAndBytesConfig
2
  import torch
3
  from PIL import Image
 
1
+ import subprocess
2
+ import sys
3
+
4
+ def install(package):
5
+ subprocess.check_call([sys.executable, "-m", "pip", "install", package])
6
+
7
+ install("evaluate")
8
+ install("jiwer")
9
+ install("--upgrade huggingface_hub")
10
+ install("gradio==3.36.0")
11
+ install("bitsandbytes")
12
+ install("git+https://github.com/huggingface/transformers.git")
13
+ install("git+https://github.com/huggingface/peft.git")
14
+ install("git+https://github.com/huggingface/accelerate.git")
15
+ install("einops")
16
+ install("safetensors")
17
+ install("torch")
18
+ install("xformers")
19
+ install("datasets")
20
+ install("stable-diffusion")
21
+
22
  from transformers import AutoProcessor, AutoModelForCausalLM, BitsAndBytesConfig
23
  import torch
24
  from PIL import Image