Spaces:
Running
Running
Update app.py
Browse files
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-
|
| 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.
|