Create app.py

app.py

@@ -0,0 +1,64 @@
+import spaces
+import gradio as gr
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+# 1. Load your model at the root level (required for ZeroGPU)
+model_id = "erdemozkan/YOLO-7B-Qwen-Coder"
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+model = AutoModelForCausalLM.from_pretrained(
+    model_id, 
+    torch_dtype=torch.bfloat16, 
+    device_map="auto"
+)
+
+# 2. The GPU-decorated function
+@spaces.GPU(duration=60) # Declares a 60s window on the H200
+def yoco_heal(broken_code):
+    prompt = f"### System: You are YOLO CODER. Analyze the following broken code and provide a self-healed, working version.\n\n### Broken Code:\n{broken_code}\n\n### Fixed Code:"
+    
+    inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
+    
+    with torch.no_grad():
+        outputs = model.generate(
+            **inputs, 
+            max_new_tokens=1024, 
+            temperature=0.7,
+            do_sample=True
+        )
+    
+    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    # Extract only the code part from the response
+    if "### Fixed Code:" in response:
+        return response.split("### Fixed Code:")[1].strip()
+    return response
+
+# 3. The "YOLO" Themed UI
+yoco_theme = gr.themes.Soft(
+    primary_hue="yellow",
+    secondary_hue="zinc",
+    neutral_hue="slate",
+).set(
+    body_background_fill="#09090b", # Dark OLED black
+    block_background_fill="#18181b",
+    button_primary_background_fill="*primary_500",
+    button_primary_text_color="black",
+)
+
+with gr.Blocks(theme=yoco_theme) as demo:
+    gr.Markdown("# 🚀 YOLO CODER: Self-Healing AI")
+    gr.Markdown("Drop your broken code. Let **yoco** fix it while you grab a coffee.")
+    
+    with gr.Row():
+        with gr.Column():
+            input_code = gr.Code(label="Paste Broken Code", language="python", lines=15)
+            heal_btn = gr.Button("YOLO IT!", variant="primary")
+        with gr.Column():
+            output_code = gr.Code(label="Self-Healed Output", language="python", lines=15)
+
+    heal_btn.click(fn=yoco_heal, inputs=input_code, outputs=output_code)
+    
+    gr.Markdown("---")
+    gr.Markdown("Built by [Erdem Ozkan](https://huggingface.co/erdemozkan) | Powered by Qwen 7B & ZeroGPU")
+
+demo.launch()