import gradio as grfrom transformers import AutoTokenizer, AutoModelForCausalLMimport torch MODEL_ID = "vishnuamarapu/Full-Fine-Tuning-Qwen-2.5-0.5B-instruct-sft" tokenizer = AutoTokenizer.from_pretrained(MODEL_ID) model = AutoModelForCausalLM.from_pretrained( MODEL_ID, torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32, device_map="auto") def chat(message, history): messages = [ { "role": "system", "content": ( "You are Vishnu's personal AI assistant. " "Answer questions about Vishnu using the provided information." ) } ] for user, assistant in history: messages.append({"role": "user", "content": user}) messages.append({"role": "assistant", "content": assistant}) messages.append({"role": "user", "content": message}) text = tokenizer.apply_chat_template( messages, tokenize=False, add_generation_prompt=True ) inputs = tokenizer( text, return_tensors="pt" ).to(model.device) outputs = model.generate( **inputs, max_new_tokens=256 ) response = tokenizer.decode( outputs[0][inputs.input_ids.shape[-1]:], skip_special_tokens=True ) return response gr.ChatInterface(chat).launch()