Spaces:
Paused
Paused
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"]
|