raghualgt commited on
Commit
6f375db
·
verified ·
1 Parent(s): 91e17f5

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -0
app.py ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ generator = pipeline(
5
+ "text-generation",
6
+ model="deepseek-ai/deepseek-coder-1.3b-instruct"
7
+ )
8
+
9
+ def generate_code(prompt):
10
+ result = generator(prompt, max_length=200)
11
+ return result[0]["generated_text"]
12
+
13
+ demo = gr.Interface(
14
+ fn=generate_code,
15
+ inputs="text",
16
+ outputs="text",
17
+ title="Python Coding Assistant",
18
+ description="Ask Python coding questions"
19
+ )
20
+
21
+ demo.launch()