LOOFYYLO commited on
Commit
32f5adb
·
verified ·
1 Parent(s): 2f84fe5

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. load_model.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  import torch
3
- from app.load_model import load_model_and_tokenizer
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 .model import FuturesModel, CustomTokenizer, build_vocabulary
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',