Omnibus commited on
Commit
b07d35a
1 Parent(s): 7ecc702

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,6 +1,6 @@
1
  from huggingface_hub import InferenceClient
2
- cl = InferenceClient("vikhyatk/moondream2")
3
- print(cl)
4
  from __future__ import annotations
5
 
6
  import torch
@@ -1166,8 +1166,8 @@ from transformers import TextIteratorStreamer
1166
  import hashlib
1167
  import os
1168
 
1169
- model_path = snapshot_download("vikhyatk/moondream1")
1170
-
1171
  vision_encoder = VisionEncoder(model_path).to(DEVICE, dtype=DTYPE)
1172
  text_model = TextModel(model_path).to(DEVICE, dtype=DTYPE)
1173
 
 
1
  from huggingface_hub import InferenceClient
2
+ #cl = InferenceClient("vikhyatk/moondream2")
3
+ #print(cl)
4
  from __future__ import annotations
5
 
6
  import torch
 
1166
  import hashlib
1167
  import os
1168
 
1169
+ #model_path = snapshot_download("vikhyatk/moondream1")
1170
+ model_path = snapshot_download("vikhyatk/moondream2")
1171
  vision_encoder = VisionEncoder(model_path).to(DEVICE, dtype=DTYPE)
1172
  text_model = TextModel(model_path).to(DEVICE, dtype=DTYPE)
1173