Spaces:
Sleeping
Sleeping
Stéphanie Kamgnia Wonkap
commited on
Commit
•
781a06e
1
Parent(s):
b99886d
fixing import
Browse files
app.py
CHANGED
@@ -3,6 +3,7 @@ import streamlit as st
|
|
3 |
import os
|
4 |
import yaml
|
5 |
from dotenv import load_dotenv
|
|
|
6 |
from src.generator import answer_with_rag
|
7 |
from ragatouille import RAGPretrainedModel
|
8 |
from src.data_preparation import split_documents
|
|
|
3 |
import os
|
4 |
import yaml
|
5 |
from dotenv import load_dotenv
|
6 |
+
import torch
|
7 |
from src.generator import answer_with_rag
|
8 |
from ragatouille import RAGPretrainedModel
|
9 |
from src.data_preparation import split_documents
|