Spaces:

AndreaAlessandrelli4
/

AvvoChat_v02

Runtime error

AndreaAlessandrelli4 commited on Jul 22, 2024

Commit

5aeefb9

verified ·

1 Parent(s): 6dc02f1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,6 +5,8 @@ from typing import Iterator
 import gradio as gr
 import spaces
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 MAX_MAX_NEW_TOKENS = 2048

 import gradio as gr
 import spaces
 import torch
+import weaviate
+from sentence_transformers import SentenceTransformer
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 MAX_MAX_NEW_TOKENS = 2048