Update main.py

main.py

@@ -64,7 +64,7 @@ def search(query: str, k: int) -> tuple[str]:
 tokenizer = transformers.AutoTokenizer.from_pretrained("Qwen/Qwen2-0.5B-Instruct")
 streamer = transformers.TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
 chatmodel = transformers.AutoModelForCausalLM.from_pretrained(
-    "Qwen/Qwen2-0.5B-Instruct"
+    "Qwen/Qwen2-0.5B-Instruct",
     torch_dtype="auto",
     device_map="auto"
 )