Spaces:

abdullahalioo
/

aiapp

Sleeping

abdullahalioo commited on Apr 26

Commit

560244c

verified ·

1 Parent(s): e8eb792

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -16,7 +16,7 @@ app.add_middleware(
 )
 # Load model and tokenizer (do this once at startup)
-model_name = "Qwen/Qwen2.5-VL-7B-Instruct"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(
     model_name,

 )
 # Load model and tokenizer (do this once at startup)
+model_name = "Qwen/Qwen2.5-7B-Instruct-1M"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(
     model_name,