CAlamos commited on
Commit
2bb120f
β€’
1 Parent(s): 019a8f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import os
4
 
5
  # Load your model
6
  print("Current working directory:", os.getcwd())
7
- model = LudwigModel.load('CAlamos/finetuned_mistral7b_title_generation')
8
 
9
  def generate_title(abstract):
10
  # Prepare the input in the format Ludwig expects
 
4
 
5
  # Load your model
6
  print("Current working directory:", os.getcwd())
7
+ model = LudwigModel.load('model_weights')
8
 
9
  def generate_title(abstract):
10
  # Prepare the input in the format Ludwig expects