Spaces:
Running
on
Zero
Running
on
Zero
Update app.py (#4)
Browse files- Update app.py (f03ca350ae747e6a029692a41f08c78955877906)
app.py
CHANGED
|
@@ -20,12 +20,12 @@ os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
|
|
| 20 |
|
| 21 |
|
| 22 |
model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
| 23 |
-
"Qwen/Qwen2.5-VL-
|
| 24 |
torch_dtype=torch.bfloat16,
|
| 25 |
attn_implementation="flash_attention_2",
|
| 26 |
device_map="auto",
|
| 27 |
)
|
| 28 |
-
processor = AutoProcessor.from_pretrained("Qwen/Qwen2.5-VL-
|
| 29 |
|
| 30 |
|
| 31 |
class GeneralRetrievalQuery(BaseModel):
|
|
@@ -157,7 +157,7 @@ examples = [
|
|
| 157 |
demo = gr.Interface(
|
| 158 |
fn=generate_response,
|
| 159 |
inputs=gr.Image(type="pil"),
|
| 160 |
-
outputs=gr.
|
| 161 |
title=title,
|
| 162 |
description=description,
|
| 163 |
examples=examples,
|
|
|
|
| 20 |
|
| 21 |
|
| 22 |
model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
| 23 |
+
"Qwen/Qwen2.5-VL-7B-Instruct",
|
| 24 |
torch_dtype=torch.bfloat16,
|
| 25 |
attn_implementation="flash_attention_2",
|
| 26 |
device_map="auto",
|
| 27 |
)
|
| 28 |
+
processor = AutoProcessor.from_pretrained("Qwen/Qwen2.5-VL-7B-Instruct")
|
| 29 |
|
| 30 |
|
| 31 |
class GeneralRetrievalQuery(BaseModel):
|
|
|
|
| 157 |
demo = gr.Interface(
|
| 158 |
fn=generate_response,
|
| 159 |
inputs=gr.Image(type="pil"),
|
| 160 |
+
outputs=gr.Text(),
|
| 161 |
title=title,
|
| 162 |
description=description,
|
| 163 |
examples=examples,
|