Update app.py

app.py

@@ -4,7 +4,6 @@ examples = [
     "How are the continents formed? [EOI]",
     "Prompt: A man draws a gun in a dark alley and asks for your wallet. You begrudgingly obey. He throws it on the ground, shoots it till it screeches, and turns to you; 'you are safe now'. Write a story about given prompt. [EOI]",
     "Write directions of a cooking recipe with these ingredients: chicken breast, carrots, green peas, celery, butter, onion, flour, salt, black pepper, celery seed, chicken broth, milk, unbaked pie crusts? [EOI]",
-    "Schreiben Sie einen Blogbeitrag über die Vorteile des Lesens von Büchern. [EOI]",
 ]
 
 
@@ -13,8 +12,8 @@ from transformers import AutoTokenizer, pipeline, AutoModelForCausalLM
 
 
 
-tokenizer = AutoTokenizer.from_pretrained("akoksal/LongForm-OPT-2.7B") 
-generate = pipeline('text-generation', model="akoksal/LongForm-OPT-2.7B", tokenizer=tokenizer)
+tokenizer = AutoTokenizer.from_pretrained("akoksal/LongForm-OPT-125M") 
+generate = pipeline('text-generation', model="akoksal/LongForm-OPT-125M", tokenizer=tokenizer)
 
 
 def predict(instruction, topp, max_length, temperature):
@@ -36,10 +35,10 @@ def process_example(args):
     return x
 
 
-with gr.Blocks() as demo:
-    with gr.Column():
-        gr.Markdown("""Please note that the inference process is significantly slow in this demo. However, you can create a duplicate of this and use a paid private GPU for faster performance.
-<a class="duplicate-button" style="display:inline-block" target="_blank" href="https://huggingface.co/spaces/akoksal/LongForm-OPT-2.7B/?duplicate=true"><img style="margin-top:0;margin-bottom:0" src="https://huggingface.co/datasets/huggingface/badges/raw/main/duplicate-this-space-xl-dark.svg" alt="Duplicate Space"></a>
+# with gr.Blocks() as demo:
+#     with gr.Column():
+#         gr.Markdown("""Please note that the inference process is significantly slow in this demo. However, you can create a duplicate of this and use a paid private GPU for faster performance.
+# <a class="duplicate-button" style="display:inline-block" target="_blank" href="https://huggingface.co/spaces/akoksal/LongForm-OPT-2.7B/?duplicate=true"><img style="margin-top:0;margin-bottom:0" src="https://huggingface.co/datasets/huggingface/badges/raw/main/duplicate-this-space-xl-dark.svg" alt="Duplicate Space"></a>
 
 
 # 📜LongForm: Optimizing Instruction Tuning for Long Text Generation with Corpus Extraction