Omnibus commited on
Commit
5e6314f
1 Parent(s): b07d35a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -1,6 +1,4 @@
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 +1164,7 @@ from transformers import TextIteratorStreamer
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
 
 
1
  from huggingface_hub import InferenceClient
 
 
2
  from __future__ import annotations
3
 
4
  import torch
 
1164
  import hashlib
1165
  import os
1166
 
1167
+ model_path = snapshot_download("vikhyatk/moondream1")
 
1168
  vision_encoder = VisionEncoder(model_path).to(DEVICE, dtype=DTYPE)
1169
  text_model = TextModel(model_path).to(DEVICE, dtype=DTYPE)
1170