Spaces:

brendon-ai
/

faq-huggingface-model

Sleeping

brendon-ai commited on Jul 4

Commit

6b52a00

verified ·

1 Parent(s): 49c1216

Update src/RAGSample.py

Files changed (1) hide show

src/RAGSample.py CHANGED Viewed

@@ -17,6 +17,7 @@ from sklearn.metrics.pairwise import cosine_similarity
 import pandas as pd
 from typing import Optional, List
 import re
 # OPTION 1: Use Hugging Face Pipeline (Recommended for HF Spaces)
 from transformers import pipeline

 import pandas as pd
 from typing import Optional, List
 import re
+import torch
 # OPTION 1: Use Hugging Face Pipeline (Recommended for HF Spaces)
 from transformers import pipeline