Update app.py
Browse files
app.py
CHANGED
|
@@ -9,6 +9,7 @@ import torch
|
|
| 9 |
import gradio as gr
|
| 10 |
import uuid
|
| 11 |
from sentence_transformers import SentenceTransformer
|
|
|
|
| 12 |
|
| 13 |
# model_name = 'google/flan-t5-base'
|
| 14 |
# model = T5ForConditionalGeneration.from_pretrained(model_name, device_map='auto', offload_folder="offload")
|
|
|
|
| 9 |
import gradio as gr
|
| 10 |
import uuid
|
| 11 |
from sentence_transformers import SentenceTransformer
|
| 12 |
+
import os
|
| 13 |
|
| 14 |
# model_name = 'google/flan-t5-base'
|
| 15 |
# model = T5ForConditionalGeneration.from_pretrained(model_name, device_map='auto', offload_folder="offload")
|