Stéphanie Kamgnia Wonkap commited on
Commit
781a06e
1 Parent(s): b99886d

fixing import

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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