Update app.py

app.py

@@ -246,6 +246,7 @@ app = gr.Interface(
 
 ####
 from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig, TextIteratorStreamer
+from threading import Thread
 
 model_id = "philschmid/instruct-igel-001"
 model = AutoModelForCausalLM.from_pretrained(model_id, low_cpu_mem_usage=True)
@@ -335,7 +336,7 @@ def generate(instruction, temperature=1.0, max_new_tokens=256, top_p=0.9, length
 ###
 classifier = pipeline("zero-shot-classification")
 text = "This is a tutorial about Hugging Face."
-candidate_labels = ["inform", "sell", "beschweren"]
+candidate_labels = ["informieren", "kaufen", "beschweren", "verkaufen"]
 
 def topic_sale_inform (text):
   res = classifier(text, candidate_labels)