Update app.py

app.py

@@ -0,0 +1,26 @@
+import gradio as gr
+from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
+
+# Load the model
+model_id = "deepseek-ai/deepseek-coder-1.3b-base"
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+model = AutoModelForCausalLM.from_pretrained(
+    model_id,
+    torch_dtype=torch.float16,
+    device_map="auto"
+)
+
+def generate_code(prompt):
+    input_ids = tokenizer(prompt, return_tensors="pt").input_ids.cuda()
+    outputs = model.generate(input_ids, max_new_tokens=300)
+    code = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return code
+
+gr.Interface(
+    fn=generate_code,
+    inputs=gr.Textbox(lines=4, label="Describe what code you want"),
+    outputs=gr.Code(label="Generated Code"),
+    title="Text to Code Generator",
+    description="Describe what you want in natural language, and get code!"
+).launch()