Update app.py
Browse files
app.py
CHANGED
|
@@ -22,7 +22,7 @@ load_dotenv(find_dotenv())
|
|
| 22 |
|
| 23 |
st.set_page_config(layout="wide")
|
| 24 |
|
| 25 |
-
checkpoint = "LaMini-T5-
|
| 26 |
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
|
| 27 |
base_model = AutoModelForSeq2SeqLM.from_pretrained(
|
| 28 |
checkpoint,
|
|
@@ -53,7 +53,8 @@ def data_ingestion():
|
|
| 53 |
def llm_pipeline():
|
| 54 |
pipe = pipeline(
|
| 55 |
'text2text-generation',
|
| 56 |
-
model = base_model,
|
|
|
|
| 57 |
tokenizer = tokenizer,
|
| 58 |
max_length = 256,
|
| 59 |
do_sample = True,
|
|
|
|
| 22 |
|
| 23 |
st.set_page_config(layout="wide")
|
| 24 |
|
| 25 |
+
checkpoint = "MBZUAI/LaMini-T5-738M"
|
| 26 |
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
|
| 27 |
base_model = AutoModelForSeq2SeqLM.from_pretrained(
|
| 28 |
checkpoint,
|
|
|
|
| 53 |
def llm_pipeline():
|
| 54 |
pipe = pipeline(
|
| 55 |
'text2text-generation',
|
| 56 |
+
#model = base_model,
|
| 57 |
+
#model = "MBZUAI/LaMini-T5-738M",
|
| 58 |
tokenizer = tokenizer,
|
| 59 |
max_length = 256,
|
| 60 |
do_sample = True,
|