Vipplav commited on
Commit
e710c18
·
verified ·
1 Parent(s): 87431a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  import torch, re
3
  from pymongo import MongoClient
4
  from datetime import datetime
5
- from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
6
  from sentence_transformers import SentenceTransformer, util
7
  from langchain_core.prompts import PromptTemplate
8
  from langchain_community.llms import HuggingFacePipeline
 
2
  import torch, re
3
  from pymongo import MongoClient
4
  from datetime import datetime
5
+ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline,AutoModelForSeq2SeqLM
6
  from sentence_transformers import SentenceTransformer, util
7
  from langchain_core.prompts import PromptTemplate
8
  from langchain_community.llms import HuggingFacePipeline