Spaces:

Loewolf
/

Loewolf-Chat

Runtime error

Loewolf commited on Nov 13, 2023

Commit

4e0cba7

1 Parent(s): 2b796ca

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,8 +3,8 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer
 import torch
 # Initialisierung des Modells und des Tokenizers
-tokenizer = GPT2Tokenizer.from_pretrained("Loewolf/GPT_1.5")
-model = GPT2LMHeadModel.from_pretrained("Loewolf/GPT_1.5")
 def generate_text(prompt):
     input_ids = tokenizer.encode(prompt, return_tensors="pt")

 import torch
 # Initialisierung des Modells und des Tokenizers
+tokenizer = GPT2Tokenizer.from_pretrained("Loewolf/GPT_1.2")
+model = GPT2LMHeadModel.from_pretrained("Loewolf/GPT_1.2")
 def generate_text(prompt):
     input_ids = tokenizer.encode(prompt, return_tensors="pt")