dwb2023 commited on
Commit
cc04f60
1 Parent(s): 88a0be3

Update app.py

Browse files

update imports

Files changed (1) hide show
  1. app.py +1 -2
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, LlavaMistralForCausalLM, LlavaNextForConditionalGeneration
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"],