Spaces:
Paused
Paused
Update app.py
Browse filesupdate imports
app.py
CHANGED
@@ -3,7 +3,7 @@ import os
|
|
3 |
import torch, torchvision, einops
|
4 |
import spaces
|
5 |
import subprocess
|
6 |
-
from transformers import AutoModelForCausalLM, AutoModel, AutoModelForVision2Seq, PaliGemmaForConditionalGeneration, LlavaForConditionalGeneration,
|
7 |
from huggingface_hub import login
|
8 |
|
9 |
# Install required package
|
@@ -44,7 +44,6 @@ with gr.Blocks() as demo:
|
|
44 |
gr.Markdown("### Vision Models")
|
45 |
vision_examples = gr.Examples(
|
46 |
examples=[
|
47 |
-
["liuhaotian/llava-v1.6-mistral-7b"],
|
48 |
["microsoft/llava-med-v1.5-mistral-7b"],
|
49 |
["llava-hf/llava-v1.6-mistral-7b-hf"],
|
50 |
["xtuner/llava-phi-3-mini-hf"],
|
|
|
3 |
import torch, torchvision, einops
|
4 |
import spaces
|
5 |
import subprocess
|
6 |
+
from transformers import AutoModelForCausalLM, AutoModel, AutoModelForVision2Seq, PaliGemmaForConditionalGeneration, LlavaForConditionalGeneration, LlavaNextForConditionalGeneration
|
7 |
from huggingface_hub import login
|
8 |
|
9 |
# Install required package
|
|
|
44 |
gr.Markdown("### Vision Models")
|
45 |
vision_examples = gr.Examples(
|
46 |
examples=[
|
|
|
47 |
["microsoft/llava-med-v1.5-mistral-7b"],
|
48 |
["llava-hf/llava-v1.6-mistral-7b-hf"],
|
49 |
["xtuner/llava-phi-3-mini-hf"],
|