Tonic commited on
Commit
e497e22
1 Parent(s): 5acdf7e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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("microsoft/DialoGPT-medium")
6
  tokenizer.padding_side = 'left'
7
- model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
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 (DialoGPT-medium) 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 = [["How are you?"]]
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,