mayankgoyal1247/Complexity

1import gradio as gr2from transformers import T5ForConditionalGeneration, T5Tokenizer3 4# Load T5 model5model_name = "t5-small"6tokenizer = T5Tokenizer.from_pretrained(model_name)7model = T5ForConditionalGeneration.from_pretrained(model_name)8 9# Function to generate response10def solve_problem(input_text):11    input_ids = tokenizer(input_text, return_tensors="pt").input_ids12    output_ids = model.generate(input_ids)13    return tokenizer.decode(output_ids[0], skip_special_tokens=True)14 15# Gradio Interface16iface = gr.Interface(fn=solve_problem, inputs="text", outputs="text", title="AI Problem Solver")17iface.launch()