acecalisto3 commited on
Commit
10e6684
·
verified ·
1 Parent(s): f4945de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -1,14 +1,20 @@
1
  import streamlit as st
 
2
  import os
3
  import subprocess
4
- from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
5
- from huggingface_hub import HfApi
 
 
 
 
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"]