Spaces:
Sleeping
Sleeping
migueldeguzmandev
commited on
Commit
•
6519026
1
Parent(s):
4ed69c2
Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
|
|
2 |
from transformers import GPT2Tokenizer, GPT2LMHeadModel
|
3 |
|
4 |
# Load the model and tokenizer
|
5 |
-
model_name = "migueldeguzmandev/
|
6 |
tokenizer = GPT2Tokenizer.from_pretrained(model_name)
|
7 |
model = GPT2LMHeadModel.from_pretrained(model_name)
|
8 |
|
@@ -49,10 +49,10 @@ interface = gr.Interface(
|
|
49 |
gr.Slider(minimum=0.00000000000000000000001, maximum=1.0, value=0.7, step=0.1, label="Temperature"),
|
50 |
],
|
51 |
outputs=gr.Textbox(label="Model Response"),
|
52 |
-
title="Hello, I'm
|
53 |
description=(
|
54 |
"""
|
55 |
-
(Bodybuilder AI is trained with
|
56 |
"""
|
57 |
),
|
58 |
# examples=examples,
|
|
|
2 |
from transformers import GPT2Tokenizer, GPT2LMHeadModel
|
3 |
|
4 |
# Load the model and tokenizer
|
5 |
+
model_name = "migueldeguzmandev/fitness_ai"
|
6 |
tokenizer = GPT2Tokenizer.from_pretrained(model_name)
|
7 |
model = GPT2LMHeadModel.from_pretrained(model_name)
|
8 |
|
|
|
49 |
gr.Slider(minimum=0.00000000000000000000001, maximum=1.0, value=0.7, step=0.1, label="Temperature"),
|
50 |
],
|
51 |
outputs=gr.Textbox(label="Model Response"),
|
52 |
+
title="Hello, I'm Fitness AI!",
|
53 |
description=(
|
54 |
"""
|
55 |
+
(Bodybuilder AI is trained with fitness, nutrition and training themed responses to random questions.)
|
56 |
"""
|
57 |
),
|
58 |
# examples=examples,
|