Update app.py

update imports to include architecture types required by various Models

app.py

@@ -3,7 +3,7 @@ import os
 import torch, torchvision, einops
 import spaces
 import subprocess
-from transformers import AutoModelForCausalLM
+from transformers import AutoModelForCausalLM, AutoModel, AutoModelForVision2Seq, PaliGemmaForConditionalGeneration, LlavaForConditionalGeneration, LlavaMistralForCausalLM, LlavaNextForConditionalGeneration
 from huggingface_hub import login
 
 # Install required package
@@ -64,8 +64,7 @@ with gr.Blocks() as demo:
                     ["google/gemma-7b"],
                     ["microsoft/Phi-3-mini-4k-instruct"],
                     ["meta-llama/Meta-Llama-3-8B"],
-                    ["mistralai/Mistral-7B-Instruct-v0.3"],
-                    ["mistralai/Codestral-22B-v0.1"]
+                    ["mistralai/Mistral-7B-Instruct-v0.3"]
                 ],
                 inputs=textbox
             )