Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -3,7 +3,7 @@ import json
|
|
| 3 |
|
| 4 |
from transformers import pipeline
|
| 5 |
# Load the pipeline
|
| 6 |
-
tourModel = pipeline(model="
|
| 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="
|
| 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 (cbrown@manhan.co)", allow_flagging='never') as iface:
|
| 79 |
iface.launch()
|