Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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="
|
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(
|
|
|
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()
|