Hugging Face's logo

Spaces:
HuggingFaceH4
/
vlm-playground
Runtime error

App Files Community
lewtun HF Staff commited on
Commit
d433054
·
verified ·
1 Parent(s): 83a0389

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -7,6 +7,9 @@ from PIL import Image
7
  import torch
8
  import spaces
9
  import os
 
 
 
10
 
11
  MODEL_ID = os.environ.get("MODEL_ID", "llava-hf/llava-v1.6-mistral-7b-hf")
12
  REVISION = os.environ.get("MODEL_REVISION", "main")
 
7
  import torch
8
  import spaces
9
  import os
10
+ from huggingface_hub import login
11
+
12
+ login(token=os.environ["HF_TOKEN"])
13
 
14
  MODEL_ID = os.environ.get("MODEL_ID", "llava-hf/llava-v1.6-mistral-7b-hf")
15
  REVISION = os.environ.get("MODEL_REVISION", "main")