LaibaIrfan commited on
Commit
c9f32bb
·
verified ·
1 Parent(s): 842ada7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,10 +1,11 @@
1
- pip install -U bitsandbytes==0.45.3 transformers accelerate torch --no-cache-dir
 
 
2
 
3
  import gradio as gr
4
  import torch
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
  from peft import PeftModel
7
-
8
  from functools import lru_cache
9
 
10
  # Define models
 
1
+ import os
2
+ os.system("pip install -U bitsandbytes==0.45.3 transformers accelerate torch --no-cache-dir")
3
+
4
 
5
  import gradio as gr
6
  import torch
7
  from transformers import AutoTokenizer, AutoModelForCausalLM
8
  from peft import PeftModel
 
9
  from functools import lru_cache
10
 
11
  # Define models