thechaiexperiment commited on
Commit
b464cc6
·
verified ·
1 Parent(s): c7e0abb

Update medical_rag.py

Browse files
Files changed (1) hide show
  1. medical_rag.py +1 -0
medical_rag.py CHANGED
@@ -1,5 +1,6 @@
1
  from fastapi import HTTPException
2
  from pydantic import BaseModel
 
3
  import nltk
4
  from transformers import (
5
  AutoTokenizer,
 
1
  from fastapi import HTTPException
2
  from pydantic import BaseModel
3
+ import torch
4
  import nltk
5
  from transformers import (
6
  AutoTokenizer,