Spaces:

nsajadi
/

speechtoknowldge

Runtime error

nsajadi commited on Oct 15, 2023

Commit

9acd6b5

•

1 Parent(s): 2419cbf

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import streamlit as st
 import torch
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
-pipe = pipeline('chat-response', model='liuhaotian/llava-v1.5-13b', device=device )
 text = st.text_area('Enter some text here!')
 if text:

 import torch
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
+pipe = pipeline('LLaVA', model='liuhaotian/llava-v1.5-13b', device=device )
 text = st.text_area('Enter some text here!')
 if text: