added requirements
Browse files
app.py
CHANGED
@@ -1,11 +1,11 @@
|
|
1 |
import gradio as gr
|
2 |
|
3 |
# Load model directly
|
4 |
-
from transformers import
|
5 |
|
6 |
# Load the LoRA model and tokenizer
|
7 |
tokenizer = AutoTokenizer.from_pretrained("ID2223JR/lora_model")
|
8 |
-
model =
|
9 |
|
10 |
# Data storage
|
11 |
ingredients_list = []
|
|
|
1 |
import gradio as gr
|
2 |
|
3 |
# Load model directly
|
4 |
+
from transformers import AutoModel, AutoTokenizer
|
5 |
|
6 |
# Load the LoRA model and tokenizer
|
7 |
tokenizer = AutoTokenizer.from_pretrained("ID2223JR/lora_model")
|
8 |
+
model = AutoModel.from_pretrained("ID2223JR/lora_model")
|
9 |
|
10 |
# Data storage
|
11 |
ingredients_list = []
|