Hugging Face's logo

Spaces:
adi2606
/
Math_Wizard
Running

App Files Community
Math_Wizard
File size: 3,136 Bytes 
91337c3
c8a4e18
91337c3
 
c8a4e18
 
91337c3
c8a4e18
 
 
91337c3
 
 
c8a4e18
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
91337c3
fa66bf8
c8a4e18
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
 
import torch
from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
import gradio as gr

# Lightweight CPU-friendly model
model_name = "microsoft/phi-1_5"

tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)
model.generation_config = GenerationConfig.from_pretrained(model_name)
model.generation_config.pad_token_id = model.generation_config.eos_token_id


def solve_math_problem(question):
    messages = [{"role": "user", "content": question}]
    input_tensor = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt")
    input_tensor = input_tensor.to(model.device)

    with torch.no_grad():
        outputs = model.generate(input_tensor, max_new_tokens=150)

    response = tokenizer.decode(outputs[0][input_tensor.shape[1]:], skip_special_tokens=True)
    return response.strip()


with gr.Blocks(css="footer {visibility: hidden}") as demo:
    gr.Markdown("# ๐Ÿง™โ€โ™‚๏ธ Math Wizard AI")
    gr.Markdown("""
        <div style="font-size: 16px; line-height: 1.5">
        Welcome to the <b>Math Wizard</b> โ€“ your intelligent assistant for solving math problems of all kinds! <br>
        Ask anything from algebra, calculus, or even about famous mathematicians.<br><br>
        </div>
    """)

    with gr.Tabs():
        with gr.Tab("๐Ÿงฎ General Math"):
            with gr.Row():
                with gr.Column():
                    question_box = gr.Textbox(
                        label="Ask your question here:",
                        placeholder="E.g. What is the derivative of x^2 + 3x + 2?",
                        lines=3
                    )
                    submit_btn = gr.Button("๐Ÿ” Solve Now")
                    clear_btn = gr.Button("โŒ Clear")

                with gr.Column():
                    answer_box = gr.Textbox(label="๐Ÿ“˜ Answer from the Wizard", lines=8, interactive=False)
                    copy_btn = gr.Button("๐Ÿ“‹ Copy Answer")

            submit_btn.click(fn=solve_math_problem, inputs=question_box, outputs=answer_box)
            clear_btn.click(lambda: ("", ""), outputs=[question_box, answer_box])
            copy_btn.click(lambda x: x, inputs=answer_box, outputs=answer_box, show_progress=False)

        with gr.Tab("๐Ÿง  Examples & Inspiration"):
            gr.Markdown("""
            <h4>Try asking things like:</h4>
            <ul>
            <li>๐Ÿงฉ Solve the equation xยฒ + 2x - 3 = 0</li>
            <li>๐Ÿ“ What is the Pythagorean theorem?</li>
            <li>๐Ÿ“Š Integrate sin(x) from 0 to ฯ€</li>
            <li>๐Ÿงฎ Tell me about Euclid</li>
            </ul>
            """)

        with gr.Tab("๐Ÿ“š About"):
            gr.Markdown("""
            <h4>About Math Wizard</h4>
            <p>This assistant is powered by a lightweight AI model that runs smoothly even on CPUs.</p>
            <p>Built with โค๏ธ using Gradio + HuggingFace Transformers</p>
            <p>Model: <code>microsoft/phi-1_5</code> optimized for reasoning and small footprint.</p>
            """)

if __name__ == "__main__":
    demo.launch()