Colby commited on
Commit
f87096b
·
verified ·
1 Parent(s): 95ae577

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import json
3
 
4
  from transformers import pipeline
5
  # Load the pipeline
6
- tourModel = pipeline(model="ywangmanhanco/finetuned-starplanner")
7
 
8
  def getTour(income,size,years,sex,edu,wrk):
9
  # person = a dict with person-level and hh-level attributes:
@@ -75,5 +75,5 @@ with gr.Interface(fn=getTour, inputs=[
75
  gr.Dropdown(["unknown", "male", "female"], label="Gender/sex"),
76
  gr.Dropdown(["unknown", "grade school","highschool", "associates", "bachelors", "graduate"], label="Educational attainment level"),
77
  gr.Dropdown(["unknown", "yes","no"], label="Worker status")],
78
- outputs=["json"], title="StarPlanner (aka GPT-Travel)", description="Authors: Colby Brown ([email protected]) and Ya Wang ([email protected]), Manhan", allow_flagging='never') as iface:
79
  iface.launch()
 
3
 
4
  from transformers import pipeline
5
  # Load the pipeline
6
+ tourModel = pipeline(model="Colby/tiny-starplanner")
7
 
8
  def getTour(income,size,years,sex,edu,wrk):
9
  # person = a dict with person-level and hh-level attributes:
 
75
  gr.Dropdown(["unknown", "male", "female"], label="Gender/sex"),
76
  gr.Dropdown(["unknown", "grade school","highschool", "associates", "bachelors", "graduate"], label="Educational attainment level"),
77
  gr.Dropdown(["unknown", "yes","no"], label="Worker status")],
78
+ outputs=["json"], title="StarPlanner (aka GPT-Travel)", description="Author: Colby Brown ([email protected])", allow_flagging='never') as iface:
79
  iface.launch()