Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from transformers import pipeline | |
| chatbot = pipeline("text-generation", model="Qwen/Qwen2.5-0.5B-Instruct", trust_remote_code=True) | |
| def respond(message, history): | |
| """Simplified response function""" | |
| prompt = "" | |
| for msg in history: | |
| prompt += f"{msg['role'].capitalize()}: {msg['content']}\n" | |
| prompt += f"Human: {message}\nAssistant:" | |
| response = chatbot(prompt, max_new_tokens=100)[0]['generated_text'] | |
| return response.split("Assistant:")[-1].strip() | |
| # Simple interface with voice | |
| with gr.Blocks() as demo: | |
| gr.Markdown("# 🎤 Voice AI Assistant") | |
| # Chat interface | |
| chat_interface = gr.ChatInterface( | |
| fn=respond, | |
| title="AI Chat", | |
| description="Type or use voice input" | |
| ) | |
| # Add voice input | |
| with gr.Row(): | |
| audio = gr.Audio(sources=["microphone"], type="filepath", label="Voice Input") | |
| demo.launch() |