Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,9 +4,8 @@ import torch
|
|
| 4 |
from PIL import Image
|
| 5 |
from transformers import MllamaForConditionalGeneration, AutoProcessor
|
| 6 |
from huggingface_hub import login
|
| 7 |
-
login()
|
| 8 |
HF_TOKEN=st.secrets["newfinegrained"]
|
| 9 |
-
|
| 10 |
def load_model_and_processor(model_id):
|
| 11 |
"""Load the model and processor."""
|
| 12 |
model = MllamaForConditionalGeneration.from_pretrained(
|
|
|
|
| 4 |
from PIL import Image
|
| 5 |
from transformers import MllamaForConditionalGeneration, AutoProcessor
|
| 6 |
from huggingface_hub import login
|
|
|
|
| 7 |
HF_TOKEN=st.secrets["newfinegrained"]
|
| 8 |
+
login(HF_TOKEN)
|
| 9 |
def load_model_and_processor(model_id):
|
| 10 |
"""Load the model and processor."""
|
| 11 |
model = MllamaForConditionalGeneration.from_pretrained(
|