Erfan11 commited on
Commit
e3f75e9
1 Parent(s): b58127a

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -21,7 +21,7 @@ def predict(text, model, tokenizer):
21
  return outputs
22
 
23
  def main():
24
- model_path = os.getenv('MODEL_PATH')
25
  model = load_model(model_path)
26
  tokenizer = load_tokenizer(model_path)
27
  # Example usage
 
21
  return outputs
22
 
23
  def main():
24
+ model_path = os.getenv('Erfan11/Neuracraft')
25
  model = load_model(model_path)
26
  tokenizer = load_tokenizer(model_path)
27
  # Example usage