change model

app.py

@@ -2,7 +2,8 @@ import gradio as gr
 from transformers import pipeline
 
 # pipeline = pipeline("text-generation")
-pipeline = pipeline("text-generation", model="TheBloke/WizardLM-Uncensored-SuperCOT-StoryTelling-30B-SuperHOT-8K-GPTQ", trust_remote_code=True)
+# pipeline = pipeline("text-generation", model="TheBloke/WizardLM-Uncensored-SuperCOT-StoryTelling-30B-SuperHOT-8K-GPTQ", trust_remote_code=True)
+pipeline = pipeline("text-generation", model="isarth/distill_gpt2_story_generator")
 
 def text_generate(input_text):
     result = pipeline(input_text)