Spaces:
Sleeping
Sleeping
acecalisto3
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -1,14 +1,20 @@
|
|
1 |
import streamlit as st
|
|
|
2 |
import os
|
3 |
import subprocess
|
4 |
-
|
5 |
-
from
|
|
|
|
|
|
|
|
|
6 |
|
7 |
# Access Hugging Face API key from secrets
|
8 |
hf_token = st.secrets["huggingface"]["hf_token"]
|
9 |
if not hf_token:
|
10 |
st.error("Hugging Face API key not found. Please make sure it is set in the secrets.")
|
11 |
|
|
|
12 |
PROJECT_ROOT = "projects"
|
13 |
AGENT_DIRECTORY = "agents"
|
14 |
AVAILABLE_CODE_GENERATIVE_MODELS = ["bigcode/starcoder", "Salesforce/codegen-350M-mono", "microsoft/CodeGPT-small"]
|
|
|
1 |
import streamlit as st
|
2 |
+
from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer, RagRetriever, AutoModelForSeq2SeqLM
|
3 |
import os
|
4 |
import subprocess
|
5 |
+
import black
|
6 |
+
from pylint import lint
|
7 |
+
from io import StringIO
|
8 |
+
import sys
|
9 |
+
import torch
|
10 |
+
from huggingface_hub import hf_hub_url, cached_download, HfApi
|
11 |
|
12 |
# Access Hugging Face API key from secrets
|
13 |
hf_token = st.secrets["huggingface"]["hf_token"]
|
14 |
if not hf_token:
|
15 |
st.error("Hugging Face API key not found. Please make sure it is set in the secrets.")
|
16 |
|
17 |
+
HUGGING_FACE_REPO_URL = "https://huggingface.co/spaces/acecalisto3/DevToolKit"
|
18 |
PROJECT_ROOT = "projects"
|
19 |
AGENT_DIRECTORY = "agents"
|
20 |
AVAILABLE_CODE_GENERATIVE_MODELS = ["bigcode/starcoder", "Salesforce/codegen-350M-mono", "microsoft/CodeGPT-small"]
|