Spaces:
Running
on
T4
Running
on
T4
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 ([email protected])", allow_flagging='never') as iface:
|
79 |
iface.launch()
|