Hugging Face's logo

Spaces:
Youssefk
/
StoryAI
Runtime error

App Files Community
Youssefk commited on
Commit
d11da8a
·
1 Parent(s): 24075a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import requests
4
  from transformers import AutoModelWithLMHead, AutoTokenizer
5
 
6
  tokenizer = AutoTokenizer.from_pretrained('microsoft/DialoGPT-small')
7
- model = AutoModelWithLMHead.from_pretrained('output-small-save')
8
 
9
  st.set_page_config(
10
  page_title="COVID Doctor using DialoGPT",
 
4
  from transformers import AutoModelWithLMHead, AutoTokenizer
5
 
6
  tokenizer = AutoTokenizer.from_pretrained('microsoft/DialoGPT-small')
7
+ model = AutoModelWithLMHead.from_pretrained('model.py')
8
 
9
  st.set_page_config(
10
  page_title="COVID Doctor using DialoGPT",