Illia56 commited on
Commit
9bf9831
1 Parent(s): 3be9889

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -56,6 +56,10 @@ openai/whisper-large-v3
56
  """
57
 
58
  models = {
 
 
 
 
59
  "Llama2-13b": {
60
  "model_link": "https://huggingface.co/meta-llama/Llama-2-13b",
61
  "chat_link": "https://huggingface-projects-llama-2-13b-chat.hf.space/",
@@ -87,7 +91,7 @@ def transcribe_video(youtube_url: str, path: str) -> List[Document]:
87
  return [Document(page_content=result[1], metadata=dict(page=1))]
88
 
89
  def predict(
90
- message: str, system_prompt: str = "", model_url: str = models["Llama2-13b"]["chat_link"]
91
  ) -> Any:
92
  """
93
  Predict a response using a client.
 
56
  """
57
 
58
  models = {
59
+ "Llama2-70b": {
60
+ "model_link": "https://huggingface.co/meta-llama/Llama-2-70b",
61
+ "chat_link": "https://ysharma-explore-llamav2-with-tgi.hf.space/",
62
+ },
63
  "Llama2-13b": {
64
  "model_link": "https://huggingface.co/meta-llama/Llama-2-13b",
65
  "chat_link": "https://huggingface-projects-llama-2-13b-chat.hf.space/",
 
91
  return [Document(page_content=result[1], metadata=dict(page=1))]
92
 
93
  def predict(
94
+ message: str, system_prompt: str = "", model_url: str = models["Llama2-70b"]["chat_link"]
95
  ) -> Any:
96
  """
97
  Predict a response using a client.