TEST / app.py
Reality123b's picture
Update app.py
b5425af verified
raw
history blame
922 Bytes
import gradio as gr
def custom_model_input(input_text):
# First prompt to be used before anything else
system_prompt = "You are a helpful assistant. Please process the user's input after this message."
# Combine system prompt with user input
prompt = f"{system_prompt} User's input: {input_text}"
# Call your model with the combined prompt
response = your_model_function(prompt) # Replace this with actual model call
return response
with gr.Blocks() as interface:
# Define the inputs and outputs for the interface
input_text = gr.Textbox()
output_text = gr.Textbox()
# Set up the function to be called when the user submits input
input_text.submit(custom_model_input, inputs=input_text, outputs=output_text)
# Load the model when the interface is initialized
interface.load("models/Qwen/QwQ-32B-Preview")
# Start the interface
interface.launch()