Upload app.py

app.py

@@ -27,7 +27,7 @@ wiki_wiki = wikipediaapi.Wikipedia('Organika ([email protected])', 'en')
 # Use a pipeline as a high-level helper
 from transformers import pipeline
 topic_model = pipeline("zero-shot-classification", model="valhalla/distilbart-mnli-12-9")
-model = pipeline("text-generation", model="Colby/StarCoder-3B-WoW-JSON")
+model = pipeline("text-generation", model="Colby/StarCoder-3B-WoW-JSON", device=0)
 
 def merlin_chat(message, history):
     chat_text = ""