Spaces:
Sleeping
Sleeping
Add the model
Browse files- README.md +13 -11
- app.py +40 -63
- modelfile +10 -0
- requirements.txt +2 -1
README.md
CHANGED
@@ -1,11 +1,13 @@
|
|
1 |
-
|
2 |
-
|
3 |
-
|
4 |
-
|
5 |
-
|
6 |
-
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
|
11 |
-
|
|
|
|
|
|
1 |
+
# Code Assistant
|
2 |
+
|
3 |
+
A multi-programming code assistant using CodeLlama (LLAMA2) and Gradio.
|
4 |
+
|
5 |
+
## How to use
|
6 |
+
|
7 |
+
- Enter your code in the input box.
|
8 |
+
- The assistant will generate and display the code output.
|
9 |
+
|
10 |
+
## Dependencies
|
11 |
+
|
12 |
+
- gradio
|
13 |
+
- langchain
|
app.py
CHANGED
@@ -1,63 +1,40 @@
|
|
1 |
-
import
|
2 |
-
|
3 |
-
|
4 |
-
|
5 |
-
|
6 |
-
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
-
|
23 |
-
|
24 |
-
|
25 |
-
|
26 |
-
|
27 |
-
|
28 |
-
|
29 |
-
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
-
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
-
|
38 |
-
|
39 |
-
|
40 |
-
|
41 |
-
|
42 |
-
"""
|
43 |
-
For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
|
44 |
-
"""
|
45 |
-
demo = gr.ChatInterface(
|
46 |
-
respond,
|
47 |
-
additional_inputs=[
|
48 |
-
gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
|
49 |
-
gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
|
50 |
-
gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
|
51 |
-
gr.Slider(
|
52 |
-
minimum=0.1,
|
53 |
-
maximum=1.0,
|
54 |
-
value=0.95,
|
55 |
-
step=0.05,
|
56 |
-
label="Top-p (nucleus sampling)",
|
57 |
-
),
|
58 |
-
],
|
59 |
-
)
|
60 |
-
|
61 |
-
|
62 |
-
if __name__ == "__main__":
|
63 |
-
demo.launch()
|
|
|
1 |
+
import requests
|
2 |
+
import json
|
3 |
+
import gradio as gr
|
4 |
+
|
5 |
+
url="http://localhost:11434/api/generate"
|
6 |
+
|
7 |
+
headers={
|
8 |
+
|
9 |
+
'Content-Type':'application/json'
|
10 |
+
}
|
11 |
+
|
12 |
+
history=[]
|
13 |
+
|
14 |
+
def generate_response(prompt):
|
15 |
+
history.append(prompt)
|
16 |
+
final_prompt="\n".join(history)
|
17 |
+
|
18 |
+
data={
|
19 |
+
"model":"codingsensei",
|
20 |
+
"prompt":final_prompt,
|
21 |
+
"stream":False
|
22 |
+
}
|
23 |
+
|
24 |
+
response=requests.post(url,headers=headers,data=json.dumps(data))
|
25 |
+
|
26 |
+
if response.status_code==200:
|
27 |
+
response=response.text
|
28 |
+
data=json.loads(response)
|
29 |
+
actual_response=data['response']
|
30 |
+
return actual_response
|
31 |
+
else:
|
32 |
+
print("error:",response.text)
|
33 |
+
|
34 |
+
|
35 |
+
interface=gr.Interface(
|
36 |
+
fn=generate_response,
|
37 |
+
inputs=gr.Textbox(lines=4,placeholder="Enter your Prompt"),
|
38 |
+
outputs="text"
|
39 |
+
)
|
40 |
+
interface.launch()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
modelfile
ADDED
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
FROM codellama
|
2 |
+
|
3 |
+
## set the temperature
|
4 |
+
PARAMETER temperature 1
|
5 |
+
|
6 |
+
## set the system prompt
|
7 |
+
SYSTEM """
|
8 |
+
You are a code teaching assistant named as codingsensei created by Kailash. Answer all the code related questions being asked.
|
9 |
+
|
10 |
+
"""
|
requirements.txt
CHANGED
@@ -1 +1,2 @@
|
|
1 |
-
|
|
|
|
1 |
+
langchain
|
2 |
+
gradio
|