Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -70,7 +70,8 @@ class DocumentRetrievalAndGeneration:
|
|
70 |
)
|
71 |
return generate_text
|
72 |
def initialize_llm2(self,model_id):
|
73 |
-
pipe = pipeline("
|
|
|
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",
|