Update app.py

app.py

@@ -6,18 +6,16 @@ import gradio as gr
 import torch
 from utils import *
 from presets import *
+import time
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
 
-#antwort=""
 ######################################################################
 #Modelle und Tokenizer
 
-#Hugging Chat nutzen
-# Create a chatbot connection
-#chatbot = hugchat.ChatBot(cookie_path="cookies.json")
-
 #Alternativ mit beliebigen Modellen:
 #base_model = "project-baize/baize-v2-7b"
-base_model = "EleutherAI/gpt-neo-1.3B"
+base_model = "HuggingFaceH4/starchat-alpha"
 tokenizer,model,device = load_tokenizer_and_model(base_model)