Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -7,8 +7,8 @@ import spaces
|
|
7 |
|
8 |
# Load the Pixtral model and processor
|
9 |
model_id = "mrcuddle/lumimaid-v0.2-8b-pixtral"
|
10 |
-
processor = AutoProcessor.from_pretrained(model_id
|
11 |
-
model = LlavaForConditionalGeneration.from_pretrained(model_id).to("cuda")
|
12 |
|
13 |
@spaces.GPU
|
14 |
def generate_text(input_text, image_url):
|
|
|
7 |
|
8 |
# Load the Pixtral model and processor
|
9 |
model_id = "mrcuddle/lumimaid-v0.2-8b-pixtral"
|
10 |
+
processor = AutoProcessor.from_pretrained(model_id)
|
11 |
+
model = LlavaForConditionalGeneration.from_pretrained(model_id, ignore_mismatched_sizes=True).to("cuda")
|
12 |
|
13 |
@spaces.GPU
|
14 |
def generate_text(input_text, image_url):
|