Colby commited on
Commit
b97fa02
·
verified ·
1 Parent(s): 03c5aaf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,7 +30,7 @@ wiki_wiki = wikipediaapi.Wikipedia('Organika ([email protected])', 'en')
30
  # Use a pipeline as a high-level helper
31
  from transformers import pipeline
32
  topic_model = pipeline("zero-shot-classification", model="valhalla/distilbart-mnli-12-9", device=0)
33
- model = pipeline("text-generation", model="Colby/StarCoder-1B-WoW-JSON", device=0)
34
 
35
  def merlin_chat(message, history):
36
  chat_text = ""
@@ -113,4 +113,4 @@ def merlin_chat(message, history):
113
  return message['content']
114
  return "🤔"
115
 
116
- gr.ChatInterface(merlin_chat).launch(share=True)
 
30
  # Use a pipeline as a high-level helper
31
  from transformers import pipeline
32
  topic_model = pipeline("zero-shot-classification", model="valhalla/distilbart-mnli-12-9", device=0)
33
+ model = pipeline("text-generation", model="Organika/StarCoder-1B-WoW-JSON", device=0)
34
 
35
  def merlin_chat(message, history):
36
  chat_text = ""
 
113
  return message['content']
114
  return "🤔"
115
 
116
+ gr.ChatInterface(merlin_chat).launch()