Spaces:

kykybeepbopboop
/

HfapiGradioTest

Build error

kykybeepbopboop commited on Apr 20, 2025

Commit

5528cd5

verified ·

1 Parent(s): a8cea06

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
-generator = pipeline("text-generation", model="mosaicml/mpt-7b-storywriter")
 def generate_text(prompt):
     result = generator(prompt, max_new_tokens=500)[0]['generated_text']

 import gradio as gr
 from transformers import pipeline
+generator = pipeline("text-generation", model="distilgpt2")
 def generate_text(prompt):
     result = generator(prompt, max_new_tokens=500)[0]['generated_text']