Madhuri123 commited on
Commit
b4c3b08
·
verified ·
1 Parent(s): 4c1be0d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -4,7 +4,9 @@ import torch
4
  import requests
5
  from PIL import Image
6
  from transformers import MllamaForConditionalGeneration, AutoProcessor
7
- !pip install 'accelerate>=0.26.0'
 
 
8
 
9
  HF_TOKEN=st.secrets["hf_token"]
10
  # Load the model and pipeline
 
4
  import requests
5
  from PIL import Image
6
  from transformers import MllamaForConditionalGeneration, AutoProcessor
7
+ import subprocess
8
+ subprocess.run(["pip", "install", "accelerate>=0.26.0"])
9
+
10
 
11
  HF_TOKEN=st.secrets["hf_token"]
12
  # Load the model and pipeline