Spaces:
Sleeping
Sleeping
Upload folder using huggingface_hub
Browse files- app.py +1 -1
- load_model.py +1 -1
app.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
import torch
|
| 3 |
-
from
|
| 4 |
|
| 5 |
# --- 1. Load Model and Tokenizer ---
|
| 6 |
# This is done once when the Gradio app starts.
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
import torch
|
| 3 |
+
from load_model import load_model_and_tokenizer
|
| 4 |
|
| 5 |
# --- 1. Load Model and Tokenizer ---
|
| 6 |
# This is done once when the Gradio app starts.
|
load_model.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
import torch
|
| 2 |
-
from
|
| 3 |
|
| 4 |
def load_model_and_tokenizer(
|
| 5 |
model_path='app/checkpoint_best.pt',
|
|
|
|
| 1 |
import torch
|
| 2 |
+
from model import FuturesModel, CustomTokenizer, build_vocabulary
|
| 3 |
|
| 4 |
def load_model_and_tokenizer(
|
| 5 |
model_path='app/checkpoint_best.pt',
|