Create app.py

app.py

@@ -0,0 +1,22 @@
+import gradio as gr
+from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
+
+# Load Mixtral model
+model_name = "mistralai/Mixtral-8x7B-Instruct-v0.1"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
+
+def solve_problem(prompt):
+    inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
+    output = model.generate(**inputs, max_length=500)
+    return tokenizer.decode(output[0])
+
+# Create Gradio UI
+iface = gr.Interface(fn=solve_problem, 
+                     inputs=gr.Textbox(lines=2, placeholder="Enter a JEE/NEET question..."), 
+                     outputs="text",
+                     title="Prof. Cool - JEE/NEET Solver",
+                     description="Ask a complex JEE Advanced or NEET question and get a detailed solution.")
+
+iface.launch()