Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -2,9 +2,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
2 |
import gradio as gr
|
3 |
import torch
|
4 |
|
5 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
6 |
tokenizer.padding_side = 'left'
|
7 |
-
model = AutoModelForCausalLM.from_pretrained("
|
8 |
|
9 |
class ChatBot:
|
10 |
def __init__(self):
|
@@ -23,8 +23,8 @@ class ChatBot:
|
|
23 |
bot = ChatBot()
|
24 |
|
25 |
title = "👋🏻Welcome to Tonic's EZ Chat🚀"
|
26 |
-
description = "You can use this Space to test out the current model (
|
27 |
-
examples = [["
|
28 |
|
29 |
iface = gr.Interface(
|
30 |
fn=bot.predict,
|
|
|
2 |
import gradio as gr
|
3 |
import torch
|
4 |
|
5 |
+
tokenizer = AutoTokenizer.from_pretrained("Tonic/mistralmed")
|
6 |
tokenizer.padding_side = 'left'
|
7 |
+
model = AutoModelForCausalLM.from_pretrained("Tonic/mistralmed")
|
8 |
|
9 |
class ChatBot:
|
10 |
def __init__(self):
|
|
|
23 |
bot = ChatBot()
|
24 |
|
25 |
title = "👋🏻Welcome to Tonic's EZ Chat🚀"
|
26 |
+
description = "You can use this Space to test out the current model (MistralMed) or duplicate this Space and use it for any other model on 🤗HuggingFace. Join me on [Discord](https://discord.gg/fpEPNZGsbt) to build together."
|
27 |
+
examples = [["What is the boiling point of nitrogen?"]]
|
28 |
|
29 |
iface = gr.Interface(
|
30 |
fn=bot.predict,
|