Update app.py
Browse files
app.py
CHANGED
@@ -53,9 +53,8 @@ def generate_response(input_text):
|
|
53 |
# Modell használata a válasz generálásához
|
54 |
outputs = model(**inputs)
|
55 |
|
56 |
-
# Válasz
|
57 |
response = tokenizer.decode(outputs.logits.argmax(dim=1)[0])
|
58 |
-
response = response.decode('utf-8', errors='replace')
|
59 |
return response
|
60 |
|
61 |
# Dokumentum feltöltése drag and drop segítségével
|
|
|
53 |
# Modell használata a válasz generálásához
|
54 |
outputs = model(**inputs)
|
55 |
|
56 |
+
# Válasz visszaadása
|
57 |
response = tokenizer.decode(outputs.logits.argmax(dim=1)[0])
|
|
|
58 |
return response
|
59 |
|
60 |
# Dokumentum feltöltése drag and drop segítségével
|