test model TheBloke/storytime-13B-GPTQ

app.py

@@ -1,9 +1,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="isarth/distill_gpt2_story_generator")
+# pipeline = pipeline("text-generation", model="isarth/distill_gpt2_story_generator")
+pipeline = pipeline("text-generation", model="TheBloke/storytime-13B-GPTQ")
 
 def text_generate(input_text):
     result = pipeline(input_text)