skip original prompt

README.md

@@ -1,6 +1,6 @@
 ---
 title: Google Gemma 7b
-emoji: 🏃
+emoji: 🚀
 colorFrom: purple
 colorTo: gray
 sdk: gradio

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
-
+import os
 from transformers import AutoTokenizer, AutoModelForCausalLM, TextStreamer
-
-tokenizer = AutoTokenizer.from_pretrained("google/gemma-7b")
-model = AutoModelForCausalLM.from_pretrained("google/gemma-7b")
-streamer = TextStreamer(tokenizer)
+token = os.environ["HF_TOKEN"]
+tokenizer = AutoTokenizer.from_pretrained("google/gemma-7b",token=token)
+model = AutoModelForCausalLM.from_pretrained("google/gemma-7b",token=token)
+streamer = TextStreamer(tokenizer,skip_prompt=True)
 
 
 def generate(inputs):