liuhaotian commited on
Commit
7bb9c1e
1 Parent(s): 14f034e
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +4 -1
app.py CHANGED
@@ -60,7 +60,7 @@ if __name__ == "__main__":
60
 
61
  print(f"args: {gws.args}")
62
 
63
- model_path = "liuhaotian/llava-v1.6-mistral-7b"
64
  bits = int(os.getenv("bits", 4))
65
  concurrency_count = int(os.getenv("concurrency_count", 5))
66
 
 
60
 
61
  print(f"args: {gws.args}")
62
 
63
+ model_path = os.getenv("model", "liuhaotian/llava-v1.6-mistral-7b")
64
  bits = int(os.getenv("bits", 4))
65
  concurrency_count = int(os.getenv("concurrency_count", 5))
66
 
requirements.txt CHANGED
@@ -1,2 +1,5 @@
1
  llava-torch==1.2.1.post1
2
- protobuf==4.23.3
 
 
 
 
1
  llava-torch==1.2.1.post1
2
+ protobuf==4.23.3
3
+
4
+ gradio==3.35.2
5
+ gradio_client==0.2.9