asterixix commited on
Commit
75dd735
1 Parent(s): 85086c2

Update app.py

Browse files

v.0.1.3
- correction of libraries

Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,9 +1,10 @@
 
 
 
1
  import streamlit as st
2
  import torch
3
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
4
- from langchain import HuggingFacePipeline
5
- from langchain_core.prompts import ChatPromptTemplate, HumanMessagePromptTemplate, SystemMessage
6
- from langchain_core.messages import SystemMessage
7
  import nltk
8
  import json
9
  import pandas as pd
 
1
+ from langchain import HuggingFacePipeline
2
+ from langchain_core.prompts import ChatPromptTemplate, HumanMessagePromptTemplate
3
+ from langchain.schema import SystemMessage
4
  import streamlit as st
5
  import torch
6
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
7
+ from langchain_community.llms.huggingface_pipeline import HuggingFacePipeline
 
 
8
  import nltk
9
  import json
10
  import pandas as pd