Update app.py

app.py

@@ -42,8 +42,32 @@ def generate_code(prompt):
     response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
     return response
 
-# Example usage (optional - remove for Spaces deployment)
-if __name__ == "__main__":
-    prompt = "write a quick sort algorithm."
-    generated_code = generate_code(prompt)
-    print(generated_code)
+with gr.Blocks(title="Qwen 14b") as demo:  # Updated title
+    with gr.Tab("Code Chat"):
+        run_button = gr.Button("Run", scale=0)
+        prompt = gr.Text(
+            label="Prompt",
+            show_label=False,
+            max_lines=1,
+            placeholder="Enter your prompt",
+            container=False,
+        )
+        result = gr.Text(
+            label="Result",
+            show_label=False,
+            max_lines=100,
+            container=False,
+        )
+        gr.on(
+            triggers=[
+                run_button_30.click,
+            ],
+      #  api_name="generate",  # Add this line
+            fn=generate_30,
+            inputs=[
+                prompt,
+            ],
+            outputs=[result],
+        )
+
+demo.launch(share=False)