Siri23 commited on
Commit
7d57e58
·
verified ·
1 Parent(s): 5f21677

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+
3
+ # Load the model
4
+ gpt_model = pipeline('text-generation', model='distilgpt2')
5
+
6
+ # Generate a workout plan
7
+ prompt = "Generate a 1-day workout plan for weight loss."
8
+ workout_plan = gpt_model(prompt, max_length=150, num_return_sequences=1)[0]['generated_text']