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,
|