Hugging Face's logo

Spaces:
Saurabh1207
/
VLM
Running

App Files Community
Saurabh Kumar commited on
Commit
7ca46d0
·
verified ·
1 Parent(s): eed28b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import streamlit as st
4
  import torch
5
  from PIL import Image
6
 
7
- @st.cache
8
  # default: Load the model on the available device(s)
9
  model = Qwen2VLForConditionalGeneration.from_pretrained(
10
  "Qwen/Qwen2-VL-7B-Instruct", torch_dtype="auto", device_map="auto"
 
4
  import torch
5
  from PIL import Image
6
 
7
+ @st.cache_resource
8
  # default: Load the model on the available device(s)
9
  model = Qwen2VLForConditionalGeneration.from_pretrained(
10
  "Qwen/Qwen2-VL-7B-Instruct", torch_dtype="auto", device_map="auto"