Spaces:
Runtime error
Runtime error
lint corrected
Browse files
app.py
CHANGED
|
@@ -11,10 +11,7 @@ import fitz # PyMuPDF
|
|
| 11 |
|
| 12 |
# model and tokenizer loading
|
| 13 |
checkpoint = "MBZUAI/LaMini-Flan-T5-248M"
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
# Model and tokenizer loading
|
| 17 |
-
# checkpoint = "model/google-flan-t5-base"
|
| 18 |
tokenizer = T5Tokenizer.from_pretrained(checkpoint)
|
| 19 |
base_model = T5ForConditionalGeneration.from_pretrained(checkpoint, device_map='auto', torch_dtype=torch.float32)
|
| 20 |
|
|
|
|
| 11 |
|
| 12 |
# model and tokenizer loading
|
| 13 |
checkpoint = "MBZUAI/LaMini-Flan-T5-248M"
|
| 14 |
+
# checkpoint = "google/flan-t5-base"
|
|
|
|
|
|
|
|
|
|
| 15 |
tokenizer = T5Tokenizer.from_pretrained(checkpoint)
|
| 16 |
base_model = T5ForConditionalGeneration.from_pretrained(checkpoint, device_map='auto', torch_dtype=torch.float32)
|
| 17 |
|