manojpatil commited on
Commit
b0184f6
β€’
1 Parent(s): 131a740

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -23,7 +23,7 @@ from langchain.vectorstores import Milvus
23
  from langchain.tools import DuckDuckGoSearchRun
24
  import requests
25
 
26
- BASE_DIR = os.getcwd()
27
 
28
  #replicate api token
29
  os.environ["REPLICATE_API_TOKEN"] = "r8_30xo4KYovs74WNJiDFmZFENUcoXUBJa1B0nat"
@@ -56,6 +56,12 @@ if response.status_code == 200:
56
  else:
57
  print("Failed to download the file.")
58
 
 
 
 
 
 
 
59
  #intialize replicate llm
60
  llm = Replicate(
61
  model="a16z-infra/llama13b-v2-chat:df7690f1994d94e96ad9d568eac121aecf50684a0b0963b25a41cc40061269e5",
@@ -247,6 +253,10 @@ async def health():
247
  """Check the api is running"""
248
  return {"status": "πŸ€™"}
249
 
 
 
 
 
250
 
251
  if __name__ == "__main__":
252
  uvicorn.run(
 
23
  from langchain.tools import DuckDuckGoSearchRun
24
  import requests
25
 
26
+
27
 
28
  #replicate api token
29
  os.environ["REPLICATE_API_TOKEN"] = "r8_30xo4KYovs74WNJiDFmZFENUcoXUBJa1B0nat"
 
56
  else:
57
  print("Failed to download the file.")
58
 
59
+ BASE_DIR = os.getcwd()
60
+ items = os.listdir(BASE_DIR)
61
+
62
+ # Print the list of items
63
+ for item in items:
64
+ print(item)
65
  #intialize replicate llm
66
  llm = Replicate(
67
  model="a16z-infra/llama13b-v2-chat:df7690f1994d94e96ad9d568eac121aecf50684a0b0963b25a41cc40061269e5",
 
253
  """Check the api is running"""
254
  return {"status": "πŸ€™"}
255
 
256
+ @app.get("/")
257
+ async def welcome():
258
+ """Welcome to pipeline 1"""
259
+ return {"status": "Welcome to pipeline 1"}
260
 
261
  if __name__ == "__main__":
262
  uvicorn.run(