Update app.py

app.py

@@ -1,10 +1,24 @@
+pip install accelerate
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import transformers
 import torch
+import streamlit as st
 
 model = "tiiuae/falcon-7b"
 
 tokenizer = AutoTokenizer.from_pretrained(model)
+text_input= st.text_area("Enter some text: ")
+    if text_input: 
+        sequences = pipeline(
+        text_input,
+        max_length=200,
+        do_sample=True,
+        top_k=10,
+        num_return_sequences=1,
+        eos_token_id=tokenizer.eos_token_id,
+    )
+    for seq in sequences:
+        print(f"Result: {seq['generated_text']}")
 pipeline = transformers.pipeline(
     "text-generation",
     model=model,
@@ -13,13 +27,5 @@ pipeline = transformers.pipeline(
     trust_remote_code=True,
     device_map="auto",
 )
-sequences = pipeline(
-   "Girafatron is obsessed with giraffes, the most glorious animal on the face of this Earth. Giraftron believes all other animals are irrelevant when compared to the glorious majesty of the giraffe.\nDaniel: Hello, Girafatron!\nGirafatron:",
-    max_length=200,
-    do_sample=True,
-    top_k=10,
-    num_return_sequences=1,
-    eos_token_id=tokenizer.eos_token_id,
-)
-for seq in sequences:
-    print(f"Result: {seq['generated_text']}")
+
+