YashVrdhan/LeetCode_Helper

1import gradio as gr2from transformers import pipeline3 4# Load the Llama model5generator = pipeline(6    "text-generation", 7    model="meta-llama/Llama-2-7b-chat-hf", 8    max_new_tokens=2009)10 11def generate_response(prompt):12    result = generator(prompt, do_sample=True)[0]['generated_text']13    return result14 15# Gradio interface16iface = gr.Interface(17    fn=generate_response, 18    inputs="text", 19    outputs="text", 20    title="Llama-2 LeetCode Helper"21)22 23iface.launch()