update'

main.py

@@ -13,7 +13,7 @@ logging.basicConfig(level=logging.INFO)
 app = FastAPI()
 
 # Qwen2.5-VL Model Setup
-qwen_checkpoint = "Qwen/Qwen2.5-VL-3B-Instruct"
+qwen_checkpoint = "Qwen/Qwen2.5-VL-7B-Instruct"
 min_pixels = 256 * 28 * 28
 max_pixels = 1280 * 28 * 28
 
@@ -30,11 +30,12 @@ qwen_model = AutoModelForCausalLM.from_pretrained(
 )
 
 # LLaMA Model Setup
-llama_model_name = "path/to/llama-uncensored-model"
+llama_model_name = "Orenguteng/Llama-3.1-8B-Lexi-Uncensored-V2"
 llama_tokenizer = AutoTokenizer.from_pretrained(llama_model_name)
 llama_model = AutoModelForCausalLM.from_pretrained(
     llama_model_name, torch_dtype=torch.float16, device_map="auto"
 )
+ 
 
 @app.get("/")
 def read_root():