abhika-m commited on
Commit
b624fdb
1 Parent(s): 69fbf92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,6 +1,9 @@
1
  import vllm
2
  import torch
3
  import gradio
 
 
 
4
 
5
  # Fava prompt
6
  INPUT = "Read the following references:\n{evidence}\nPlease identify all the errors in the following text using the information in the references provided and suggest edits if necessary:\n[Text] {output}\n[Edited] "
 
1
  import vllm
2
  import torch
3
  import gradio
4
+ from transformers import whoami
5
+
6
+ user = whoami(token=HF_TOKEN)
7
 
8
  # Fava prompt
9
  INPUT = "Read the following references:\n{evidence}\nPlease identify all the errors in the following text using the information in the references provided and suggest edits if necessary:\n[Text] {output}\n[Edited] "