aizanlabs commited on
Commit
6e0bec2
·
verified ·
1 Parent(s): 17fef7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -70,7 +70,8 @@ class DocumentRetrievalAndGeneration:
70
  )
71
  return generate_text
72
  def initialize_llm2(self,model_id):
73
- pipe = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.2")
 
74
  # model_name = "mistralai/Mistral-7B-Instruct-v0.2"
75
  # pipeline = transformers.pipeline(
76
  # "text-generation",
 
70
  )
71
  return generate_text
72
  def initialize_llm2(self,model_id):
73
+ pipe = pipeline("text2text-generation", model="google/flan-t5-small")
74
+ # pipe = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.2")
75
  # model_name = "mistralai/Mistral-7B-Instruct-v0.2"
76
  # pipeline = transformers.pipeline(
77
  # "text-generation",